def test_two_epoch(self):
model = models.PiecewiseConstantSize(50, (10, 100))
self.assertEqual(model.population_configurations[0].initial_size, 50)
self.assertEqual(len(model.demographic_events), 1)
event = model.demographic_events[0]
self.assertEqual(event.time, 10)
self.assertEqual(event.initial_size, 100)
self.assertEqual(event.growth_rate, 0)
def test_two_epoch(self):
model = models.PiecewiseConstantSize(50, (10, 100))
assert model.model.populations[0].initial_size == 50
assert model.model.num_populations == 1
assert model.model.num_events == 1
event = model.model.events[0]
assert event.time == 10
assert event.initial_size == 100
assert event.growth_rate is None
def test_two_epoch(self):
model = models.PiecewiseConstantSize(50, (10, 100))
self.assertEqual(model.model.populations[0].initial_size, 50)
self.assertEqual(model.model.num_populations, 1)
self.assertEqual(model.model.num_events, 1)
event = model.model.events[0]
self.assertEqual(event.time, 10)
self.assertEqual(event.initial_size, 100)
self.assertIsNone(event.growth_rate)
def test_three_epoch(self):
model = models.PiecewiseConstantSize(0.1, (0.1, 10), (0.2, 100))
self.assertEqual(model.population_configurations[0].initial_size, 0.1)
self.assertEqual(len(model.demographic_events), 2)
event = model.demographic_events[0]
self.assertEqual(event.time, 0.1)
self.assertEqual(event.initial_size, 10)
self.assertEqual(event.growth_rate, 0)
event = model.demographic_events[1]
self.assertEqual(event.time, 0.2)
self.assertEqual(event.initial_size, 100)
self.assertEqual(event.growth_rate, 0)
def test_three_epoch(self):
model = models.PiecewiseConstantSize(0.1, (0.1, 10), (0.2, 100))
self.assertEqual(model.model.populations[0].initial_size, 0.1)
self.assertEqual(model.model.num_populations, 1)
self.assertEqual(model.model.num_events, 2)
event = model.model.events[0]
self.assertEqual(event.time, 0.1)
self.assertEqual(event.initial_size, 10)
self.assertIsNone(event.growth_rate)
event = model.model.events[1]
self.assertEqual(event.time, 0.2)
self.assertEqual(event.initial_size, 100)
self.assertIsNone(event.growth_rate)
def test_three_epoch(self):
model = models.PiecewiseConstantSize(0.1, (0.1, 10), (0.2, 100))
assert model.model.populations[0].initial_size == 0.1
assert model.model.num_populations == 1
assert model.model.num_events == 2
event = model.model.events[0]
assert event.time == 0.1
assert event.initial_size == 10
assert event.growth_rate is None
event = model.model.events[1]
assert event.time == 0.2
assert event.initial_size == 100
assert event.growth_rate is None
def test_single_epoch(self):
model = models.PiecewiseConstantSize(100)
self.assertEqual(model.population_configurations[0].initial_size, 100)
self.assertEqual(len(model.population_configurations), 1)
self.assertEqual(len(model.demographic_events), 0)
class TestTwoEpochModel(unittest.TestCase, ModelTestMixin):
model = models.PiecewiseConstantSize(100, (10, 10))
class TestConstantSizeModel(unittest.TestCase, ModelTestMixin):
model = models.PiecewiseConstantSize(100)
def test_single_epoch(self):
model = models.PiecewiseConstantSize(100)
self.assertEqual(model.model.populations[0].initial_size, 100)
self.assertEqual(model.model.num_populations, 1)
self.assertEqual(model.model.num_events, 0)
def test_single_epoch(self):
model = models.PiecewiseConstantSize(100)
assert model.model.populations[0].initial_size == 100
assert model.model.num_populations == 1
assert model.model.num_events == 0
class TestTwoEpochModel(DemographicModelTestMixin):
model = models.PiecewiseConstantSize(100, (10, 10))
class TestConstantSizeModel(DemographicModelTestMixin):
model = models.PiecewiseConstantSize(100)
