def test_restart_days(self):
rst_file = EclFile(self.test_file)
self.assertAlmostEqual(0.0, rst_file.iget_restart_sim_days(0))
self.assertAlmostEqual(31.0, rst_file.iget_restart_sim_days(1))
self.assertAlmostEqual(274.0, rst_file.iget_restart_sim_days(10))
with self.assertRaises(KeyError):
rst_file.restart_get_kw("Missing", dtime=datetime.date(2004, 1, 1))
with self.assertRaises(IndexError):
rst_file.restart_get_kw("SWAT", dtime=datetime.date(1985, 1, 1))
def test_kw(self):
f = EclFile(self.u_file)
kw1 = f["SWAT"][40]
kw2 = f.restart_get_kw("SWAT", datetime.datetime(2003, 3, 1))
kw3 = f.restart_get_kw("SWAT",
datetime.datetime(2003, 3, 1),
copy=True)
self.assertTrue(kw1.equal(kw2))
self.assertTrue(kw1.equal(kw3))
with self.assertRaises(IndexError):
kw4 = f.restart_get_kw("SWAT", datetime.datetime(2009, 3, 17))
