def test_calc_is_valid_type_error(self):
test_name = "Calc Chart Data Test #33"
data_to_test = 1234
class_to_test = CalcData()
expected_result = False
# Action
result = class_to_test.is_valid(data_to_test)
# 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_file_contents_are_valid(self):
test_name = "Calc Chart Data Test #1"
data_to_test = "A001,gender M,age 52,sales 123,bmi Overweight,"\
"salary 50,birthday 23/10/1998,valid 1,"
class_to_test = CalcData()
expected_result = True
# Action
result = class_to_test.is_valid(data_to_test)
# 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_file_contents_are_not_valid(self):
test_name = "Calc Chart Data Test #2"
data_to_test = "a001,M,52,123,Overweight,50,23-10-1998 Z005,F,"\
"18,624,Normal,85,25-04-1158 C078,F,35"
class_to_test = CalcData()
expected_result = False
# Action
result = class_to_test.is_valid(data_to_test)
# 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 get_data(self, choice):
# choose file to get data from
file_name = input("Please enter the filename to read data from >>> ")
# check file exists
try:
file_contents = self._load_file_data(file_name)
# create instance of CalcData to calc data for particular chart
i = CalcData()
if file_contents: # if contents isn't empty
if i.is_valid(file_contents): # as long as contents are valid
i.calculate(file_contents,
choice) # calculate the chart data
return i # return i so chart can be called
else:
print(Err.get_error_message(210))
except FileNotFoundError:
print(Err.ErrorHandler.get_error_message(201))
