Add test for experiment_type setter

qiskit-community · Nov 2, 2023 · af2eaee · af2eaee
1 parent 657eb1c
commit af2eaee
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test/framework/test_composite.py b/test/framework/test_composite.py
@@ -151,6 +151,21 @@ def test_roundtrip_serializable(self):
 
         self.assertRoundTripSerializable(exp)
 
+    def test_experiment_type(self):
+        """Test experiment_type setter."""
+
+        exp1 = FakeExperiment([0])
+
+        par_exp1 = ParallelExperiment([exp1], flatten_results=False)
+        batch_exp1 = BatchExperiment([exp1], flatten_results=False)
+        self.assertEqual(par_exp1.experiment_type, "ParallelExperiment")
+        self.assertEqual(batch_exp1.experiment_type, "BatchExperiment")
+
+        par_exp2 = ParallelExperiment([exp1], flatten_results=False, experiment_type="yooo")
+        batch_exp2 = BatchExperiment([exp1], flatten_results=False, experiment_type="blaaa")
+        self.assertEqual(par_exp2.experiment_type, "yooo")
+        self.assertEqual(batch_exp2.experiment_type, "blaaa")
+
 
 @ddt
 class TestCompositeExperimentData(QiskitExperimentsTestCase):