Skip to content

Commit

Permalink
modify test_SymbolicPulse
Browse files Browse the repository at this point in the history
  • Loading branch information
to24toro committed Feb 2, 2023
1 parent e2c5148 commit 72cd9ad
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions test/dynamics/pulse/test_pulse_to_signals.py
Original file line number Diff line number Diff line change
Expand Up @@ -330,7 +330,7 @@ def test_multiple_channels_with_gaps(self):
self.assertTrue(signals[3].carrier_freq == 4.0)

def test_SymbolicPulse(self):
"""Test get samples of Pulse not get_waveform but get_samples function."""
"""Test SymbolicPulse with get samples function."""
gauss_get_waveform_samples = (
pulse.Gaussian(duration=5, amp=0.983, sigma=2.0).get_waveform().samples
)
Expand All @@ -353,14 +353,6 @@ def setUp(self):
pulse.Constant(duration=5, amp=0.1).get_waveform().samples
)

def test_get_samples(self):
"""Test get samples of Pulse not get_waveform but get_samples function in Jax case."""
gauss_get_samples = get_samples(Gaussian(duration=5, amp=0.983, sigma=2.0))
self.assertTrue(isinstance(gauss_get_samples, jnp.ndarray))
self.assertAllClose(
gauss_get_samples, self.gauss_get_waveform_samples, atol=1e-7, rtol=1e-7
)

def test_jit_get_samples(self):
"""Test compiling to get samples of Pulse."""

Expand Down

0 comments on commit 72cd9ad

Please sign in to comment.