def test_sse1(self): f = lambda x: x*2 xd = np.array([1, 2, 3]) yd = np.array([2, 4, 6]) self.assertEqual(sse(f, xd, yd), 0) yd = np.array([2, 3, 3]) self.assertEqual(sse(f, xd, yd), 10)
def test_sse2(self): ''' Make sure we can mix ints and floats ''' f = lambda x: x*2 xd = np.array([1]) yd = np.array([2.5]) self.assertEqual(sse(f, xd, yd), 0.25)