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_ml_class_file(self): filename = 'ex1data2.txt' LinearRegression.load_data(filename)
def test_2pt(self): input_ = [(0, 1), (1, 0)] expected = input_ actual = LinearRegression.fit_2d(input_) self.assertEqual(expected, actual)