def test_repr_html_is_html_table_after_fit(fit_func, fit_data): model = Model(fit_func) model.fit(*fit_data) assert ('<table>' in model._repr_html_())
def test_repr_html_is_repr_before_fit(fit_func): model = Model(fit_func) assert (model._repr_html_() == model.__repr__())