adapted test

kaelyndunnell · Jan 6, 2025 · f50f771 · f50f771
1 parent 4fd3f66
commit f50f771
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/test/test_scenario.py b/test/test_scenario.py
@@ -159,21 +159,17 @@ def test_maximum_time():
     [
         (0, pulse1),
         (6000, pulse2),
-        (1e5, None),
+        (1e5, pulse2),
         (pulse1.nb_pulses * pulse1.total_duration, pulse2),
     ],
 )
 def test_get_pulse(t, expected_pulse):
 
     my_scenario = Scenario([pulse1, pulse2])
 
-    if expected_pulse is None:
-        with pytest.raises(ValueError):
-            my_scenario.get_pulse(t=t)
-    else:
-        pulse = my_scenario.get_pulse(t=t)
-        print(pulse.pulse_type)
-        assert pulse == expected_pulse
+    pulse = my_scenario.get_pulse(t=t)
+    print(pulse.pulse_type)
+    assert pulse == expected_pulse
 
 
 @pytest.mark.parametrize("t, expected_pulse", [(100, pulse1)])