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
