class EnvTest(DatabaseFixture, PEP8MixIn):
PEP8_FILES = ["golem/environments/environment.py"]
def setUp(self):
super().setUp()
self.env = Environment()
def test_get_performance(self):
# given
perf_value = 6666.6
perf = Performance(environment_id=Environment.get_id(),
value=perf_value)
perf.save()
# then
self.assertEqual(self.env.get_performance(), perf_value)
def test_get_source_code(self):
# check defaults
assert self.env.get_source_code() is None
# given
file_name = path.join(self.path, "mainprogramfile")
self.env.main_program_file = file_name
# then
assert self.env.get_source_code() is None
# re-given
with open(file_name, 'w') as f:
f.write("PROGRAM CODE")
# then
self.env.main_program_file = file_name
assert self.env.get_source_code() == "PROGRAM CODE"
def test_run_default_benchmark(self):
assert Environment.get_performance() == 0.0
assert Environment.run_default_benchmark(save=True) > 0.0
assert Environment.get_performance() > 0.0
def test_get_min_accepted_performance_default(self):
self.assertEqual(MinPerformanceMultiplier.get(), 0.0)
self.assertEqual(self.env.get_min_accepted_performance(), 0.0)
def test_get_min_accepted_performance(self):
# given
p = Performance(environment_id=Environment.get_id(),
min_accepted_step=100)
p.save()
MinPerformanceMultiplier.set(3.141)
# then
self.assertEqual(MinPerformanceMultiplier.get(), 3.141)
self.assertEqual(self.env.get_min_accepted_performance(), 314.1)
def test_get_source_code(self):
env = Environment()
assert env.get_source_code() is None
file_name = path.join(self.path, "mainprogramfile")
env.main_program_file = file_name
assert env.get_source_code() is None
with open(file_name, 'w') as f:
f.write("PROGRAM CODE")
env.main_program_file = file_name
assert env.get_source_code() == "PROGRAM CODE"