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")
