def testKeyedDataSetSaveCompressed(self):
keys = ['k1', 'k2', 'k3', '4', '5']
values = ['v1', 'v2', 100, 0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds1_file.txt.gz"
KeyedDataSetFileUtilities.saveToCompressedFile(filename, kds, "H1",
"H2")
file_contents = gzip.open(filename, 'rb').read().strip()
self.assertEqual(file_contents,
"H1 H2\nk1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromCompressedFile(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.saveToCompressedFile(filename, kds, None,
None)
file_contents = gzip.open(filename, 'r').read().strip()
self.assertEqual(file_contents, "k1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromCompressedFile(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 testKeyedDataSetSaveCompressed(self):
keys = ['k1','k2','k3','4','5']
values = ['v1','v2',100,0.4, 5]
kds = KeyedDataSet("Test1", None, values, keys)
filename = "kds1_file.txt.gz"
KeyedDataSetFileUtilities.saveToCompressedFile(filename, kds, "H1", "H2")
file_contents = gzip.open(filename, 'rb').read().strip()
self.assertEqual(file_contents,
"H1 H2\nk1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromCompressedFile(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.saveToCompressedFile(filename, kds, None, None)
file_contents = gzip.open(filename, 'r').read().strip()
self.assertEqual(file_contents, "k1 v1\nk2 v2\nk3 100\n4 0.4\n5 5")
kds = KeyedDataSetFileUtilities.loadFromCompressedFile(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)
