def test_sech_deriv(self): """Test sech derivative pulse.""" amp = 0.5 center = 20 sigma = 2 times = np.linspace(0, 40, 1000) sech_deriv_arr = continuous.sech_deriv(times, amp, center, sigma) self.assertEqual(sech_deriv_arr.dtype, np.complex_) self.assertAlmostEqual(continuous.sech_deriv(np.array([0]), amp, center, sigma)[0], 0, places=3)
def test_sech_deriv(self): """Test discrete sampled sech derivative pulse.""" amp = 0.5 sigma = 2 duration = 10 center = duration/2 times = np.arange(0, duration) + 0.5 sech_deriv_ref = continuous.sech_deriv(times, amp, center, sigma) sech_deriv_pulse = library.sech_deriv(duration, amp, sigma) self.assertIsInstance(sech_deriv_pulse, Waveform) np.testing.assert_array_almost_equal(sech_deriv_pulse.samples, sech_deriv_ref)