def test_data_table_from_file(self): d = DataTable( file_stream( DATA_FILE, contains_row_names=False ) ) self.assertEqual(len(d), NUMBER_OF_ROWS_IN_DATA_FILE) for i, cn in enumerate(d.column_names): self.assertEqual('field{}'.format(i), cn) for i, column_name in enumerate(column_names): d.column_names[i] = column_name for name1, name2 in zip(d.column_names, column_names): self.assertEqual(name1, name2)
def test_invalid_file(self): with self.assertRaises(IOError): for row in file_stream('non_existing_file'): pass
def setUp(self): Configuration.initialize(CONFIG_FILE) self.data_table = DataTable(file_stream(DATA_FILE, contains_row_names=False)) for i, column_name in enumerate(column_names): self.data_table.column_names[i] = column_name