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)