class TestPlottingFunction(object):
def test_population_has_plotting_option(self):
self.pop = Pop("test/test/parameters.txt")
self.pop.create()
try:
self.pop.launch(dev='on')
except TypeError as e:
assert False, "allow for 'on' device to show plots"
os.remove("output/vigilance_out.txt")
os.remove("output/vigilance_out.gif")
os.remove('output/resources_out.txt')
os.remove('output/exploration_out.txt')
os.remove('output/grid_out.gif')
def test_gif_is_created_only_when_dev_on(self):
self.pop = Pop("test/test/parameters.txt")
self.pop.create()
self.pop.launch(dev='on')
self.filesListRootOut = os.listdir("./output")
assert "vigilance_out.gif" in self.filesListRootOut, "no output gif created"
os.remove("output/vigilance_out.gif")
self.pop.launch()
self.filesListRootOut = os.listdir("./output")
assert "vigilance_out.gif" not in self.filesListRootOut, "should not have created output gif!"
os.remove("output/vigilance_out.txt")
os.remove('output/resources_out.txt')
os.remove('output/exploration_out.txt')
os.remove('output/grid_out.gif')
def test_grid_gif_is_created(self):
self.pop = Pop("test/test/parameters.txt")
self.pop.create()
self.pop.launch(dev='on')
self.filesListRootOut = os.listdir("./output")
assert "grid_out.gif" in self.filesListRootOut, "no output grid gif created"
os.remove("output/vigilance_out.txt")
os.remove("output/vigilance_out.gif")
os.remove('output/resources_out.txt')
os.remove('output/exploration_out.txt')
os.remove('output/grid_out.gif')
from model.population import Population as Pop
import sys
try:
visual = sys.argv[1]
except IndexError:
visual = "off"
print("Visualisation turned off. Type 'python run.py on' to run simulation with visualisation")
pop = Pop()
pop.create()
pop.launch(dev=visual)
# pop.launch(dev='on')