def test_good_math(self): ret = double(3) self.assertEqual(6, ret)
def test_four(self): ret = double(4) self.assertEqual(8, ret)
def test_two(self): ret = double(2) self.assertEqual(4, ret)
def test_double_vector(self): x = np.array([2, 5]) ret = double(x) ref = np.array([4, 10]) self.assertTrue(np.array_equal(ref, ret))