def test_02_02_add_column_with_data(self):
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
strings = ["foo", "bar"]
column = csv.add_column("kolumn", strings)
self.assertEqual(len(column), len(strings))
for s0, s1 in zip(strings, column):
self.assertEqual(s0, s1)
def test_03_01_getitem(self):
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
strings = ["foo", "bar"]
csv.add_column("kolumn", strings)
column = csv["kolumn"]
self.assertEqual(len(column), len(strings))
for s0, s1 in zip(strings, column):
self.assertEqual(s0, s1)
def test_05_01_get_column_names_etc(self):
d = {
"random": ["foo", "bar", "baz"],
"fruits": ["lemon", "cherry", "orange", "apple"],
"rocks": ["granite", "basalt", "limestone"]
}
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
csv.set_all(d)
for key in d:
self.assertIn(key, csv.get_column_names())
self.assertIn(key, csv)
self.assertIn(key, csv.keys())
self.assertIn(key, csv.iterkeys())
def test_04_01_set_all(self):
d = {
"random": ["foo", "bar", "baz"],
"fruits": ["lemon", "cherry", "orange", "apple"],
"rocks": ["granite", "basalt", "limestone"]
}
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
csv.set_all(d)
for key, strings in d.iteritems():
column = csv[key]
self.assertEqual(len(column), len(strings))
for s0, s1 in zip(strings, column):
self.assertEqual(s0, s1)
def test_02_01_add_column_empty(self):
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
column = csv.add_column("kolumn")
self.assertTrue("kolumn" in csv.top_level_group)
def test_01_02_init_twice(self):
H5DICT.HDFCSV(self.hdf_file, "csv")
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
self.assertEqual(len(csv), 0)
def test_01_01_init(self):
csv = H5DICT.HDFCSV(self.hdf_file, "csv")
self.assertEqual(len(csv), 0)