Exemplo n.º 1
0
 def test_get_feature_names(self):
     feature_names = ["f1", "f2", "f3"]
     data_set = np.array([feature_names, ["1", "2", "3"], ["", "4", "5"]])
     preprocessor = Preprocessor(data_set)
     names = preprocessor.get_feature_names()
     self.assertTrue(feature_names[0] == names[0])
     self.assertTrue(feature_names[1] == names[1])
     self.assertTrue(feature_names[2] == names[2])
Exemplo n.º 2
0
 def test_get_feature_names(self):
     feature_names = ["f1", "f2", "f3"]
     data_set = np.array([feature_names, ["1", "2", "3"], ["", "4", "5"]])
     preprocessor = Preprocessor(data_set)
     names = preprocessor.get_feature_names()
     self.assertTrue(feature_names[0] == names[0])
     self.assertTrue(feature_names[1] == names[1])
     self.assertTrue(feature_names[2] == names[2])