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)
