def test_square_root(self): test_data = CSVReader('Unit_Test_Square_Root.csv').data for row in test_data: self.assertAlmostEqual( self.calculator.square_root(float(row['Value 1'])), float(row['Result'])) self.assertAlmostEqual(self.calculator.result, float(row['Result']))
def test_subtraction(self): test_data = CSVReader('Unit_Test_Subtraction.csv').data for row in test_data: self.assertEqual( self.calculator.subtract(float(row['Value 2']), float(row['Value 1'])), float(row['Result'])) self.assertEqual(self.calculator.result, float(row['Result']))
def test_multiplication(self): test_data = CSVReader('Unit_Test_Multiplication.csv').data for row in test_data: self.assertEqual( self.calculator.multiply(float(row['Value 1']), float(row['Value 2'])), float(row['Result'])) self.assertEqual(self.calculator.result, float(row['Result']))
def test_division(self): test_data = CSVReader('Unit_Test_Division.csv').data for row in test_data: self.assertAlmostEqual( self.calculator.divide(float(row['Value 2']), float(row['Value 1'])), float(row['Result'])) self.assertAlmostEqual(self.calculator.result, float(row['Result']))
def __init__(self): print("Começo do treinamento") data = CSVReader() trainingX = data.getXTest() trainingY = data.getYTest() self.poly_features = PolynomialFeatures(degree=3) polyX = self.poly_features.fit_transform(trainingX) # fit final model self.model = LinearRegression() self.model = self.model.fit(polyX, trainingY) print("Final do Treinamento")