def test_for_fail(self):
wrong_result = np.zeros(10)
num_points = 5
list_of_t = np.linspace(-np.pi, np.pi, num_points)
num_sine_osc = 99
calculated_result = square_wave.calculate_square_wave_for(list_of_t, num_sine_osc)
self.assertFalse((wrong_result == calculated_result), "Results are equal but they should not be equal")
def test_square_wave(self):
assumed_result = np.array(
[-1.650777599717162E-14, -1.0032151693727007, 0.0, 1.0032151693727007, 1.650777599717162E-14]
)
num_points = 5
list_of_t = np.linspace(-np.pi, np.pi, num_points)
num_sine_osc = 99
calculated_result = square_wave.calculate_square_wave_for(list_of_t, num_sine_osc)
self.assertTrue(np.allclose(assumed_result, calculated_result), "Results are not equal")
