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)
