def test_calc_line_salary(self):
test_name = "Calc Chart Data Test #4"
data_to_test = "salary"
class_to_test = CalcData()
expected_result = [500]
# Action
class_to_test.calc_line(data_to_test, 500)
result = class_to_test.salary_list
# Assert
try:
self.assertTrue(result == expected_result)
except AssertionError:
print("{} Failed - Should be {}, but was {}.".format(
test_name, expected_result, result))
else:
print("{} Passed".format(test_name))
def test_line_chart_values(self):
test_name = "Calc Chart Data Line Chart Test #02"
data_to_test = None
class_to_test = CalcData()
expected_result = None
class_to_test.calc_line('age', 50)
class_to_test.calc_line('salary', 500)
# Action
result = class_to_test.line_chart()
# Assert
try:
self.assertTrue(result == expected_result)
except AssertionError:
print("{} Failed - Should be {}, but was {}.".format(
test_name, expected_result, result))
else:
print("{} Passed".format(test_name))
