Fixes unit tests

Author: AndrewSazonov

src/easydiffraction/experiments/experiment.py

@@ -182,8 +182,10 @@ def __init__(self,
         super().__init__(name=name,
                          type=type)
 
-        self._peak_profile_type = DEFAULT_PEAK_PROFILE_TYPE[self.type.scattering_type.value][self.type.beam_mode.value]
-        self._background_type = DEFAULT_BACKGROUND_TYPE
+        self._background_type: str = DEFAULT_BACKGROUND_TYPE
+        self.background = BackgroundFactory.create(
+            background_type=self.background_type)
+        self._peak_profile_type: str = DEFAULT_PEAK_PROFILE_TYPE[self.type.scattering_type.value][self.type.beam_mode.value]
         self.peak = PeakFactory.create(
             scattering_type=self.type.scattering_type.value,
             beam_mode=self.type.beam_mode.value,

tests/unit_tests/experiments/components/test_experiment_type.py

@@ -7,7 +7,8 @@ def test_experiment_type_initialization():
     experiment_type = ExperimentType(
         sample_form="powder",
         beam_mode="CW",
-        radiation_probe="neutron"
+        radiation_probe="neutron",
+        scattering_type="bragg"
     )
 
     assert isinstance(experiment_type.sample_form, Descriptor)
@@ -30,7 +31,8 @@ def test_experiment_type_properties():
     experiment_type = ExperimentType(
         sample_form="single_crystal",
         beam_mode="TOF",
-        radiation_probe="x-ray"
+        radiation_probe="x-ray",
+        scattering_type="bragg"
     )
 
     assert experiment_type.category_key == "expt_type"
@@ -45,7 +47,8 @@ def no_test_experiment_type_locking_attributes():
     experiment_type = ExperimentType(
         sample_form="powder",
         beam_mode="CW",
-        radiation_probe="neutron"
+        radiation_probe="neutron",
+        scattering_type="bragg"
     )
     experiment_type._locked = True  # Disallow adding new attributes
     experiment_type.new_attribute = "value"

tests/unit_tests/experiments/components/test_peak.py

@@ -142,5 +142,5 @@ def test_peak_factory_create_invalid_beam_mode():
 
 
 def test_peak_factory_create_invalid_profile_type():
-    with pytest.raises(ValueError, match="Unsupported profile type 'invalid' for mode 'constant wavelength'.*"):
+    with pytest.raises(ValueError, match="Unsupported profile type 'invalid' for beam mode 'constant wavelength'.*"):
         PeakFactory.create(beam_mode="constant wavelength", profile_type="invalid")

tests/unit_tests/experiments/test_experiment.py

@@ -13,7 +13,8 @@
 from easydiffraction.core.constants import (
     DEFAULT_SAMPLE_FORM,
     DEFAULT_BEAM_MODE,
-    DEFAULT_RADIATION_PROBE
+    DEFAULT_RADIATION_PROBE,
+    DEFAULT_SCATTERING_TYPE
 )
 
 
@@ -22,7 +23,8 @@ def expt_type():
     return ExperimentType(
         sample_form=DEFAULT_SAMPLE_FORM,
         beam_mode=DEFAULT_BEAM_MODE,
-        radiation_probe="xray"
+        radiation_probe='xray',
+        scattering_type='bragg'
     )
 
 
@@ -53,8 +55,9 @@ def test_powder_experiment_initialization(expt_type):
     experiment = PowderExperiment(name="PowderTest", type=expt_type)
     assert experiment.name == "PowderTest"
     assert experiment.type == expt_type
-    assert experiment.peak is not None
     assert experiment.background is not None
+    assert experiment.peak is not None
+    assert experiment.linked_phases is not None
 
 
 def test_powder_experiment_load_ascii_data(expt_type):
@@ -69,16 +72,6 @@ def test_powder_experiment_load_ascii_data(expt_type):
     assert np.array_equal(experiment.datastore.pattern.meas_su, mock_data[:, 2])
 
 
-def test_powder_experiment_show_meas_chart(expt_type):
-    experiment = PowderExperiment(name="PowderTest", type=expt_type)
-    experiment.datastore = MagicMock()
-    experiment.datastore.pattern.x = [1, 2, 3]
-    experiment.datastore.pattern.meas = [10, 20, 30]
-    with patch("easydiffraction.utils.chart_plotter.ChartPlotter.plot") as mock_plot:
-        experiment.show_meas_chart()
-        mock_plot.assert_called_once()
-
-
 def test_single_crystal_experiment_initialization(expt_type):
     experiment = ConcreteSingleCrystalExperiment(name="SingleCrystalTest", type=expt_type)
     assert experiment.name == "SingleCrystalTest"
@@ -99,6 +92,7 @@ def test_experiment_factory_create_powder():
         sample_form="powder",
         beam_mode=DEFAULT_BEAM_MODE,
         radiation_probe=DEFAULT_RADIATION_PROBE,
+        scattering_type=DEFAULT_SCATTERING_TYPE
     )
     assert isinstance(experiment, PowderExperiment)
     assert experiment.name == "PowderTest"