def test_migration_rates(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0.002, 0.003).model
assert np.shape(model.migration_matrix) == (3, 3)
assert model.migration_matrix[0][1] == 0.002
assert model.migration_matrix[1][0] == 0.003
# check these are the only two nonzero entries in the migration matrix
assert np.sum(np.array(model.migration_matrix) != 0) == 2
def test_migration_rates(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0.002, 0.003)
self.assertEqual(np.shape(model.migration_matrix), (3, 3))
self.assertEqual(model.migration_matrix[0][1], 0.002)
self.assertEqual(model.migration_matrix[1][0], 0.003)
# check these are the only two nonzero entries in the migration matrix
self.assertEqual(np.sum(np.array(model.migration_matrix) != 0), 2)
def test_split(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0, 0)
self.assertEqual(len(model.demographic_events), 2)
self.assertEqual(model.demographic_events[0].time, 50)
self.assertEqual(model.demographic_events[1].time, 50)
def test_pop_configs(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0, 0)
self.assertEqual(len(model.population_configurations), 3)
self.assertEqual(model.population_configurations[0].initial_size, 200)
self.assertEqual(model.population_configurations[1].initial_size, 300)
self.assertEqual(model.population_configurations[2].initial_size, 100)
def test_split(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0, 0).model
self.assertEqual(len(model.events), 1)
self.assertEqual(model.events[0].time, 50)
def test_split(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0, 0).model
assert len(model.events) == 1
assert model.events[0].time == 50
def test_pop_configs(self):
model = models.IsolationWithMigration(100, 200, 300, 50, 0, 0).model
assert model.num_populations == 3
assert model.populations[0].initial_size == 200
assert model.populations[1].initial_size == 300
assert model.populations[2].initial_size == 100
