def test_3(self): """test framework methods""" myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) mcFw = mf.MonteCarloFramework(dynFrw, 5) self.assertTrue(myModel.sampleNumbers() == range(1, 6))
def test_4(self): """test generation of sample directories""" myModel = mcTestModels.staticModel() statFrw = sf.StaticFramework(myModel) mcFw = mf.MonteCarloFramework(statFrw, 5) directoriesCreated = True for directory in range(1, 6): if not os.path.isdir(str(directory)): directoriesCreated = False self.assertTrue(directoriesCreated) for directory in range(1, 6): shutil.rmtree(str(directory)) myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) mcFw = mf.MonteCarloFramework(dynFrw, 5) directoriesCreated = True for directory in range(1, 6): if not os.path.isdir(str(directory)): directoriesCreated = False self.assertTrue(directoriesCreated) for directory in range(1, 6): shutil.rmtree(str(directory))
def test_4(self): """test generation of sample directories""" myModel = mcTestModels.staticModel() statFrw = sf.StaticFramework(myModel) mcFw = mf.MonteCarloFramework(statFrw, 5) directoriesCreated = True for directory in range(1,6): if not os.path.isdir(str(directory)): directoriesCreated = False self.assertTrue(directoriesCreated) for directory in range(1,6): shutil.rmtree(str(directory)) myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) mcFw = mf.MonteCarloFramework(dynFrw, 5) directoriesCreated = True for directory in range(1,6): if not os.path.isdir(str(directory)): directoriesCreated = False self.assertTrue(directoriesCreated) for directory in range(1,6): shutil.rmtree(str(directory))
def test_3(self): """test framework methods""" myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) mcFw = mf.MonteCarloFramework(dynFrw, 5) self.assertTrue(myModel.sampleNumbers() == range(1,6))
def test_6(self): """test execution of sections, report, readmap for a static model""" myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) dynFrw.setQuiet(True) mcFw = mf.MonteCarloFramework(dynFrw, 5) mcFw.setQuiet(True) # see if existing directories were emptied for directory in range(1,6): assert len(os.listdir(str(directory))) == 0 mcFw.run() filesInitialCreated = True filesDynamicCreated = True filesPremcCreated = True filesPostmcCreated = True for sample in range(1,6): nameInit = "mcdi%d.map" % (sample) if not os.path.isfile(os.path.join(str(sample), nameInit)): filesInitialCreated = False for timestep in range(1,11): nameDyn = fb.generateNameT("mcdd%d" % (sample), timestep) if not os.path.isfile(os.path.join(str(sample), nameDyn)): filesDynamicCreated = False for timestep in range(1,11): namePre = "premc_%d_%d.map" % (sample, timestep) namePost ="postmc_%d_%d.map" % (sample, timestep) if not os.path.isfile(namePre): filesPremcCreated = False if not os.path.isfile(namePost): filesPostmcCreated = False self.assertTrue(filesInitialCreated) self.assertTrue(filesPremcCreated) self.assertTrue(filesPostmcCreated) self.assertTrue(filesDynamicCreated)
def test_6(self): """test execution of sections, report, readmap for a static model""" myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) dynFrw.setQuiet(True) mcFw = mf.MonteCarloFramework(dynFrw, 5) mcFw.setQuiet(True) # see if existing directories were emptied for directory in range(1, 6): assert len(os.listdir(str(directory))) == 0 mcFw.run() filesInitialCreated = True filesDynamicCreated = True filesPremcCreated = True filesPostmcCreated = True for sample in range(1, 6): nameInit = "mcdi%d.map" % (sample) if not os.path.isfile(os.path.join(str(sample), nameInit)): filesInitialCreated = False for timestep in range(1, 11): nameDyn = fb.generateNameT("mcdd%d" % (sample), timestep) if not os.path.isfile(os.path.join(str(sample), nameDyn)): filesDynamicCreated = False for timestep in range(1, 11): namePre = "premc_%d_%d.map" % (sample, timestep) namePost = "postmc_%d_%d.map" % (sample, timestep) if not os.path.isfile(namePre): filesPremcCreated = False if not os.path.isfile(namePost): filesPostmcCreated = False self.assertTrue(filesInitialCreated) self.assertTrue(filesPremcCreated) self.assertTrue(filesPostmcCreated) self.assertTrue(filesDynamicCreated)
def test_8(self): """ test not removing sample directories """ # Create output from a previous run for directory in range(1, 6): dirname = '{}'.format(directory) shutil.rmtree(dirname) os.mkdir('{}'.format(directory)) with open(os.path.join(dirname, 'sentinel.txt'), 'w') as content: content.write('dummy text') myModel = mcTestModels.dynamicModel() dynFrw = df.DynamicFramework(myModel, 10) mcFw = mf.MonteCarloFramework(dynFrw, 5, remove_dirs=False) file_exists = True for directory in range(1, 6): dirname = '{}'.format(directory) if not os.path.exists(os.path.join(dirname, 'sentinel.txt')): file_exists = False self.assertTrue(file_exists)