def test_normal_mu_default(self): args = MagicMock(type=['normal'], mu=None) fill_default_mu(args) self.assertEqual(args.mu, [0.])
def test_poisson_mu_non_default(self): args = MagicMock(type=['normal'], mu=[7.]) fill_default_mu(args) self.assertEqual(args.mu, [7.])
def test_poisson_mu_default(self): args = MagicMock(type=['poisson'], mu=None) fill_default_mu(args) self.assertEqual(args.mu, [1.])