def test_read_only_creation(self):
with TestAreaContext("enkf/summary_key_set/read_only_write_test"):
keys = SummaryKeySet()
keys.addSummaryKey("FOPT")
keys.addSummaryKey("WWCT")
filename = "test.txt"
keys.writeToFile(filename)
keys_from_file = SummaryKeySet(filename, read_only=True)
self.assertItemsEqual(keys.keys(), keys_from_file.keys())
self.assertTrue(keys_from_file.isReadOnly())
self.assertFalse(keys_from_file.addSummaryKey("WOPR"))
def test_write_to_and_read_from_file(self):
with TestAreaContext("enkf/summary_key_set/write_test"):
keys = SummaryKeySet()
keys.addSummaryKey("FOPT")
keys.addSummaryKey("WWCT")
filename = "test.txt"
self.assertFalse(os.path.exists(filename))
keys.writeToFile(filename)
self.assertTrue(os.path.exists(filename))
keys_from_file = SummaryKeySet(filename)
self.assertItemsEqual(keys.keys(), keys_from_file.keys())
def test_creation(self):
keys = SummaryKeySet()
self.assertEqual(len(keys), 0)
self.assertTrue(keys.addSummaryKey("FOPT"))
self.assertEqual(len(keys), 1)
self.assertTrue("FOPT" in keys)
self.assertItemsEqual(["FOPT"], keys.keys())
self.assertTrue(keys.addSummaryKey("WWCT"))
self.assertEqual(len(keys), 2)
self.assertTrue("WWCT" in keys)
self.assertItemsEqual(["WWCT", "FOPT"], keys.keys())