Example #1
0
 def test_as_cleaned(self):
     csv_data_file = 'data.csv'
     raw_data = data_manager.load_raw_data(csv_data_file)
     ids, ftrs = data_manager.as_cleaned(raw_data)
     self.assertEqual(len(ids), 166)
     self.assertEqual(len(ftrs), 166)
     self.assertEqual(len(ftrs[0]), 268)
Example #2
0
 def test_as_arrays(self):
     csv_data_file = 'data.csv'
     raw_data = data_manager.load_raw_data(csv_data_file)
     ids, ftrs = data_manager.as_cleaned(raw_data)
     ids, ftrs, rm_tps, rm_tms, dl = data_manager.as_arrays(ids, ftrs)