def testKeyedDataSetSave(self):
keys = ['k1','k2','k3','4','5']
values = ['v1','v2',100,0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds1_file.txt"
KeyedDataSetFileUtilities.saveToFile(filename, kds, "H1", "H2")
with open(filename, 'r') as f_:
file_contents = f_.read().strip()
self.assertEqual(file_contents,
"H1 H2\nk1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
# test loading it back in
kds = KeyedDataSetFileUtilities.loadFromFile(filename, header="H1 H2")
self.assertEqual(kds.name, filename)
self.assertEqual(kds.index, None)
self.assertEqual(len(kds.data), 5)
self.assertEqual(kds.values_by_key["k1"], "v1")
self.assertEqual(kds.values_by_key["k2"], "v2")
self.assertEqual(kds.values_by_key["k3"], "100")
self.assertEqual(kds.values_by_key["4"], "0.4")
self.assertEqual(kds.values_by_key["5"], "5")
# No header
KeyedDataSetFileUtilities.saveToFile(filename, kds, None, None)
with open(filename, 'r') as f_:
file_contents = f_.read().strip()
self.assertEqual(file_contents, "k1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromFile(filename)
self.assertEqual(kds.name, filename)
self.assertEqual(kds.index, None)
self.assertEqual(len(kds.data), 5)
self.assertEqual(kds.values_by_key["k1"], "v1")
self.assertEqual(kds.values_by_key["k2"], "v2")
self.assertEqual(kds.values_by_key["k3"], "100")
self.assertEqual(kds.values_by_key["4"], "0.4")
self.assertEqual(kds.values_by_key["5"], "5")
os.remove(filename)
def testKeyedDataSetSave(self):
keys = ['k1','k2','k3','4','5']
values = ['v1','v2',100,0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds1_file.txt"
KeyedDataSetFileUtilities.saveToFile(filename, kds, "H1", "H2")
file_contents = open(filename, 'r').read().strip()
self.assertEqual(file_contents,
"H1 H2\nk1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
# test loading it back in
kds = KeyedDataSetFileUtilities.loadFromFile(filename, header="H1 H2")
self.assertEqual(kds.name, filename)
self.assertEqual(kds.index, None)
self.assertEqual(len(kds.data), 5)
self.assertEqual(kds.values_by_key["k1"], "v1")
self.assertEqual(kds.values_by_key["k2"], "v2")
self.assertEqual(kds.values_by_key["k3"], "100")
self.assertEqual(kds.values_by_key["4"], "0.4")
self.assertEqual(kds.values_by_key["5"], "5")
# No header
KeyedDataSetFileUtilities.saveToFile(filename, kds, None, None)
file_contents = open(filename, 'r').read().strip()
self.assertEqual(file_contents, "k1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromFile(filename)
self.assertEqual(kds.name, filename)
self.assertEqual(kds.index, None)
self.assertEqual(len(kds.data), 5)
self.assertEqual(kds.values_by_key["k1"], "v1")
self.assertEqual(kds.values_by_key["k2"], "v2")
self.assertEqual(kds.values_by_key["k3"], "100")
self.assertEqual(kds.values_by_key["4"], "0.4")
self.assertEqual(kds.values_by_key["5"], "5")
os.remove(filename)
