def test_without_gen_kw(self): case_directory = self.createTestPath('local/snake_oil_no_data/') with TestAreaContext('test_enkf_runpath', store_area=True) as work_area: work_area.copy_directory(case_directory) res_config = ResConfig('snake_oil_no_data/snake_oil_no_gen_kw.ert') main = EnKFMain(res_config) iactive = BoolVector(initial_size=main.getEnsembleSize(), default_value=False) iactive[0] = True fs = main.getEnkfFsManager().getCurrentFileSystem() run_context = main.getRunContextENSEMPLE_EXPERIMENT(fs, iactive) main.createRunpath(run_context) self.assertDirectoryExists( 'storage/snake_oil_no_gen_kw/runpath/realisation-0/iter-0') self.assertFileDoesNotExist( 'storage/snake_oil_no_gen_kw/runpath/realisation-0/iter-0/parameters.txt' ) self.assertEqual( len(os.listdir('storage/snake_oil_no_gen_kw/runpath')), 1) self.assertEqual( len( os.listdir( 'storage/snake_oil_no_gen_kw/runpath/realisation-0')), 1)
def test_with_gen_kw(self): case_directory = self.createTestPath("local/snake_oil_no_data/") with TestAreaContext("test_enkf_runpath", store_area=True) as work_area: work_area.copy_directory(case_directory) res_config = ResConfig("snake_oil_no_data/snake_oil.ert") main = EnKFMain(res_config) iactive = BoolVector(initial_size=main.getEnsembleSize(), default_value=False) iactive[0] = True fs = main.getEnkfFsManager().getCurrentFileSystem() run_context = main.getRunContextENSEMPLE_EXPERIMENT(fs, iactive) main.createRunpath(run_context) self.assertFileExists( "snake_oil_no_data/storage/snake_oil/runpath/realisation-0/iter-0/parameters.txt" ) self.assertEqual( len(os.listdir("snake_oil_no_data/storage/snake_oil/runpath")), 1) self.assertEqual( len( os.listdir( "snake_oil_no_data/storage/snake_oil/runpath/realisation-0" )), 1, ) rp = main.create_runpath_list() self.assertEqual(len(rp), 0) rp.load() self.assertEqual(len(rp), 1)
def test_with_gen_kw(copy_case): copy_case("local/snake_oil") shutil.rmtree("storage") res_config = ResConfig("snake_oil.ert") main = EnKFMain(res_config) fs = main.getEnkfFsManager().getCurrentFileSystem() run_context = main.getRunContextENSEMPLE_EXPERIMENT(fs, [True]) main.createRunpath(run_context) assert os.path.exists("storage/snake_oil/" "runpath/realization-0/iter-0/parameters.txt") assert len(os.listdir("storage/snake_oil/runpath")) == 1 assert len(os.listdir("storage/snake_oil/runpath/realization-0")) == 1 rp = main.create_runpath_list() assert len(rp) == 0 rp.load() assert len(rp), 1
def test_without_gen_kw(copy_case): copy_case("local/snake_oil") with fileinput.input("snake_oil.ert", inplace=True) as fin: for line in fin: if line.startswith("GEN_KW"): continue print(line, end="") shutil.rmtree("storage") assert "GEN_KW" not in Path("snake_oil.ert").read_text("utf-8") res_config = ResConfig("snake_oil.ert") main = EnKFMain(res_config) fs = main.getEnkfFsManager().getCurrentFileSystem() run_context = main.getRunContextENSEMPLE_EXPERIMENT(fs, [True]) main.createRunpath(run_context) assert os.path.exists("storage/snake_oil/runpath/realization-0/iter-0") assert not os.path.exists( "storage/snake_oil/runpath/realization-0/iter-0/parameters.txt") assert len(os.listdir("storage/snake_oil/runpath")) == 1 assert len(os.listdir("storage/snake_oil/runpath/realization-0")) == 1