def test_representative(self):
expected = 0
actual = step_function(-2)
self.assertEqual(expected, actual)
expected = 1
actual = step_function(4)
self.assertEqual(expected, actual)
def test_boundary_return_zero(self):
expected = 0
actual = step_function(0)
self.assertEqual(expected, actual)
def test_supports_ndarray(self):
expected = np.array([0, 0, 1])
actual = step_function(np.array([-1, 0, 1]))
np.testing.assert_allclose(expected, actual, rtol=0, atol=0)