def test_show_all_basic(clean_ui): """Set up a very basic data/model/fit""" ui.load_arrays(1, [1, 2, 4], [3, 5, 5]) ui.set_source(ui.scale1d.mdl) ui.fit() ui.conf() ui.proj() ui.covar() def get(value): out = StringIO() getattr(ui, f"show_{value}")(outfile=out) ans = out.getvalue() assert len(ans) > 1 # trim the trailing "\n" return ans[:-1] # All we are really checking is that the show_all output is the # comppsite of the following. We are not checking that the # actual output makes sense for any command. # expected = get("data") + get("model") + get("fit") + get("conf") + \ get("proj") + get("covar") got = get("all") assert expected == got
def test_covar_wrong_stat(stat, clean_ui, setup_covar): "Test an exception is thrown is the proper stat is not set" ui.covar() ui.set_stat(stat) with pytest.raises(ValueError) as exc: ui.get_draws() assert WRONG_STAT_MSG.format(stat) == str(exc.value)
def test_covar_wstat_no_background(clean_ui, setup_covar): "Test an exception is thrown when wstat is used without background" ui.covar() ui.set_stat("wstat") with pytest.raises(StatErr) as exc: ui.get_draws() assert WSTAT_ERR_MSG == str(exc.value)
def test_covar_wstat_no_background(self): ui.covar() ui.set_stat("wstat") try: ui.get_draws() except StatErr as ve: self.assertEqual(self.wstat_err_msg, str(ve)) return self.fail(self.fail_msg)
def test_covar_wstat_no_background(self): ui.covar() ui.set_stat("wstat") try: ui.get_draws() except StatErr as ve: self.assertEqual(self.wstat_err_msg, ve.message) return self.fail(self.fail_msg)
def test_covar_as_none(self): for stat in self.right_stats - {'wstat'}: ui.set_stat(stat) ui.fit() ui.covar() niter = 10 stat, accept, params = ui.get_draws(niter=niter) self.assertEqual(niter + 1, stat.size) self.assertEqual(niter + 1, accept.size) self.assertEqual((2, niter + 1), params.shape) self.assertTrue(numpy.any(accept))
def test_covar_as_none(self): for stat in self.right_stats - {'wstat'}: ui.set_stat(stat) ui.fit() ui.covar() niter = 10 stat, accept, params = ui.get_draws(niter=niter) self.assertEqual(niter+1, stat.size) self.assertEqual(niter+1, accept.size) self.assertEqual((2, niter+1), params.shape) self.assertTrue(numpy.any(accept))
def test_covar_as_none(stat, clean_ui, setup_covar): ui.set_stat(stat) ui.fit() ui.covar() niter = 10 stat, accept, params = ui.get_draws(niter=niter) n = niter + 1 assert stat.size == n assert accept.size == n assert params.shape == (2, n) assert np.any(accept)
def test_covar_wrong_stat(self): ui.covar() fail = False wrong_stats = set(ui.list_stats()) - self.right_stats for stat in wrong_stats: ui.set_stat(stat) try: ui.get_draws() except ValueError as ve: self.assertEqual(self.wrong_stat_msg.format(stat), str(ve)) continue fail = True break if fail: self.fail(self.fail_msg)
def test_covar_wrong_stat(self): ui.covar() fail = False wrong_stats = set(ui.list_stats()) - self.right_stats for stat in wrong_stats: ui.set_stat(stat) try: ui.get_draws() except ValueError as ve: self.assertEqual(self.wrong_stat_msg.format(stat), ve.message) continue fail = True break if fail: self.fail(self.fail_msg)