def testKeyedDataSetWriteContents(self):
keys = ['k1', 'k2', 'k3', '4', '5']
values = ['v1', 'v2', 100, 0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds_file.txt"
with open(filename, 'w') as file:
KeyedDataSetFileUtilities.writeContents(file, 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")
# No Header
with open(filename, 'w') as file:
KeyedDataSetFileUtilities.writeContents(file, 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")
# Test loading it back
with open(filename, 'r') as file:
kds = KeyedDataSetFileUtilities.loadContents(file, 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 testKeyedDataSetWriteContents(self):
keys = ['k1','k2','k3','4','5']
values = ['v1','v2',100,0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds_file.txt"
with open(filename, 'w') as file:
KeyedDataSetFileUtilities.writeContents(file, 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")
# No Header
with open(filename, 'w') as file:
KeyedDataSetFileUtilities.writeContents(file, 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")
# Test loading it back
with open(filename, 'r') as file:
kds = KeyedDataSetFileUtilities.loadContents(file, 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)
