add set_parsed_profile test with ProfileParser

Author: cadenmyers13

src/diffpy/srfit/fitbase/profile.py

@@ -159,7 +159,7 @@ def load_parsed_data(self, parser):
 
         This sets the xobs, yobs, dyobs arrays as well as the metadata.
         """
-        x, y, junk, dy = parser.get_data()
+        x, y, dx, dy = parser.get_data()
         self.meta = dict(parser.get_metadata())
         self.set_observed_profile(x, y, dy)
         return

tests/test_profile.py

@@ -22,6 +22,7 @@
 from numpy import allclose, arange, array, array_equal, ones_like
 
 from diffpy.srfit.exceptions import SrFitError
+from diffpy.srfit.fitbase import ProfileParser
 from diffpy.srfit.fitbase.profile import Profile
 
 
@@ -307,5 +308,20 @@ def _test(p):
     return
 
 
+def test_load_parsed_data(parser_datafiles):
+    """Test the load_parsed_data method."""
+    prof = Profile()
+    parser = ProfileParser()
+    datafile = parser_datafiles / "four_col.gr"
+    parser.parse_file(datafile)
+    prof.load_parsed_data(parser)
+    expected_xobs = [1.0, 1.1, 1.2]
+    expected_yobs = [2.0, 2.1, 2.2]
+    expected_dyobs = [0.2, 0.4, 0.6]
+    assert prof.xobs.tolist() == expected_xobs
+    assert prof.yobs.tolist() == expected_yobs
+    assert prof.dyobs.tolist() == expected_dyobs
+
+
 if __name__ == "__main__":
     unittest.main()