def test_wrong_config_objects():
osmo = Osmo(OneStepModel())
try:
osmo.test_end_condition = RandomAlgorithm()
except AttributeError:
pass
except:
raise
try:
osmo.algorithm = Length(1)
except AttributeError:
pass
except:
raise
def test_weighted_algorithm():
class HistoryTest:
steps = list()
def __init__(self):
pass
@staticmethod
def weight_first():
return 1
def step_first(self):
self.steps.append('step_first')
@staticmethod
def weight_second():
return 2
def step_second(self):
self.steps.append('step_second')
@staticmethod
def weight_third():
return 4
def step_third(self):
self.steps.append('step_third')
model = HistoryTest()
osmo = Osmo(model)
osmo.test_end_condition = Length(1)
osmo.test_end_condition = Length(100)
osmo.algorithm = WeightedAlgorithm()
osmo.generate()
step_first_count = osmo.history.get_step_count(
osmo.model.get_step_by_name("step_first"))
step_second_count = osmo.history.get_step_count(
osmo.model.get_step_by_name("step_second"))
step_third_count = osmo.history.get_step_count(
osmo.model.get_step_by_name("step_third"))
compare_first = step_first_count * (1.0 / model.weight_first())
compare_second = step_second_count * (1.0 / model.weight_second())
compare_third = step_third_count * (1.0 / model.weight_third())
assert abs(compare_first - compare_second) < 2
assert abs(compare_first - compare_third) < 2