def test_4pt(self):
input_ = [(0, 1), (0, 2), (2, 1), (2, 2)]
expected = [(0, 1.5), (1, 1.5)]
actual = LinearRegression.fit_2d(input_)
self.assertEqual(expected, actual)
def test_2pt(self):
input_ = [(0, 1), (1, 0)]
expected = input_
actual = LinearRegression.fit_2d(input_)
self.assertEqual(expected, actual)