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)