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)