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")