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