def test_summary_observation_collector(self):
with ErtTestContext("python/enkf/export/summary_observation_collector", self.config) as context:
ert = context.getErt()
self.assertTrue(SummaryObservationCollector.summaryKeyHasObservations(ert, "FOPT"))
self.assertFalse(SummaryObservationCollector.summaryKeyHasObservations(ert, "FGIR"))
keys = SummaryObservationCollector.getAllObservationKeys(ert)
self.assertTrue("FOPT" in keys)
self.assertTrue("RPR:2" in keys)
data = SummaryObservationCollector.loadObservationData(ert, "default")
self.assertFloatEqual(data["RPR:2"]["2001-01-01"], 278)
self.assertFloatEqual(data["STD_RPR:2"]["2001-01-01"], 41.7)
self.assertFloatEqual(data["FOPT"]["2000-02-01"], 619907.0)
self.assertFloatEqual(data["STD_FOPT"]["2000-02-01"], 61990.7)
with self.assertRaises(KeyError):
fgir = data["FGIR"]
data = SummaryObservationCollector.loadObservationData(ert, "default", ["RPR:3"])
self.assertFloatEqual(data["RPR:3"]["2001-01-01"], 283.5)
self.assertFloatEqual(data["STD_RPR:3"]["2001-01-01"], 2)
with self.assertRaises(KeyError):
data["FOPR"]
def test_summary_observation_collector(self):
with ErtTestContext("python/enkf/export/summary_observation_collector", self.config) as context:
ert = context.getErt()
self.assertTrue(SummaryObservationCollector.summaryKeyHasObservations(ert, "FOPR"))
self.assertFalse(SummaryObservationCollector.summaryKeyHasObservations(ert, "FOPT"))
keys = SummaryObservationCollector.getAllObservationKeys(ert)
self.assertTrue("FOPR" in keys)
self.assertTrue("WOPR:OP1" in keys)
self.assertFalse("WOPR:OP2" in keys)
data = SummaryObservationCollector.loadObservationData(ert, "default_0")
self.assertFloatEqual(data["FOPR"]["2010-01-10"], 0.001696887)
self.assertFloatEqual(data["STD_FOPR"]["2010-01-10"], 0.1)
self.assertFloatEqual(data["WOPR:OP1"]["2010-03-31"], 0.1)
self.assertFloatEqual(data["STD_WOPR:OP1"]["2010-03-31"], 0.05)
with self.assertRaises(KeyError):
fgir = data["FGIR"]
data = SummaryObservationCollector.loadObservationData(ert, "default_0", ["WOPR:OP1"])
self.assertFloatEqual(data["WOPR:OP1"]["2010-03-31"], 0.1)
self.assertFloatEqual(data["STD_WOPR:OP1"]["2010-03-31"], 0.05)
with self.assertRaises(KeyError):
data["FOPR"]
def test_summary_observation_collector(self):
with ErtTestContext("python/enkf/export/summary_observation_collector",
self.config) as context:
ert = context.getErt()
self.assertTrue(
SummaryObservationCollector.summaryKeyHasObservations(
ert, "FOPR"))
self.assertFalse(
SummaryObservationCollector.summaryKeyHasObservations(
ert, "FOPT"))
keys = SummaryObservationCollector.getAllObservationKeys(ert)
self.assertTrue("FOPR" in keys)
self.assertTrue("WOPR:OP1" in keys)
self.assertFalse("WOPR:OP2" in keys)
data = SummaryObservationCollector.loadObservationData(
ert, "default_0")
self.assertFloatEqual(data["FOPR"]["2010-01-10"], 0.001696887)
self.assertFloatEqual(data["STD_FOPR"]["2010-01-10"], 0.1)
self.assertFloatEqual(data["WOPR:OP1"]["2010-03-31"], 0.1)
self.assertFloatEqual(data["STD_WOPR:OP1"]["2010-03-31"], 0.05)
with self.assertRaises(KeyError):
fgir = data["FGIR"]
data = SummaryObservationCollector.loadObservationData(
ert, "default_0", ["WOPR:OP1"])
self.assertFloatEqual(data["WOPR:OP1"]["2010-03-31"], 0.1)
self.assertFloatEqual(data["STD_WOPR:OP1"]["2010-03-31"], 0.05)
with self.assertRaises(KeyError):
data["FOPR"]
def test_summary_observation_collector(self):
with ErtTestContext("python/enkf/export/summary_observation_collector",
self.config) as context:
ert = context.getErt()
self.assertTrue(
SummaryObservationCollector.summaryKeyHasObservations(
ert, "FOPT"))
self.assertFalse(
SummaryObservationCollector.summaryKeyHasObservations(
ert, "FGIR"))
keys = SummaryObservationCollector.getAllObservationKeys(ert)
self.assertTrue("FOPT" in keys)
self.assertTrue("RPR:2" in keys)
data = SummaryObservationCollector.loadObservationData(
ert, "default")
self.assertFloatEqual(data["RPR:2"]["2001-01-01"], 278)
self.assertFloatEqual(data["STD_RPR:2"]["2001-01-01"], 41.7)
self.assertFloatEqual(data["FOPT"]["2000-02-01"], 619907.0)
self.assertFloatEqual(data["STD_FOPT"]["2000-02-01"], 61990.7)
with self.assertRaises(KeyError):
fgir = data["FGIR"]
data = SummaryObservationCollector.loadObservationData(
ert, "default", ["RPR:3"])
self.assertFloatEqual(data["RPR:3"]["2001-01-01"], 283.5)
self.assertFloatEqual(data["STD_RPR:3"]["2001-01-01"], 2)
with self.assertRaises(KeyError):
data["FOPR"]
def gatherSummaryObservationData(ert, case, key):
if ert.getKeyManager().isKeyWithObservations(key):
return SummaryObservationCollector.loadObservationData(ert, case, [key]).dropna()
else:
return DataFrame()
