def load_file(self, file_name):
with open(file_name) as file:
self.data = file.readlines()
self.data = [x.strip() for x in self.data]
parser = Parser()
self.data = parser.parse_iris_data(self.data)
x = 0
for i in range(len(self.data)):
normalize_data = list(
map(normalize, self.data[i][0:len(self.data[i]) - 1]))
normalize_data.append(self.data[i][-1])
self.data[i] = normalize_data
if x % 2 == 0:
self.test_data.append(self.data[i])
else:
self.train_data.append(self.data[i])
x += 1
def test_iris_parser(self):
parser = Parser()
parsed_data = parser.parse_iris_data(["1,2,3,4,Iris-versicolor"])
self.assertEqual([[1.0, 2.0, 3.0, 4.0, [0, 1, 0]]], parsed_data)
