def test_repr_mimebundle_(): # Checks the display configuration flag controls the json output tree = DecisionTreeClassifier() output = tree._repr_mimebundle_() assert "text/plain" in output assert "text/html" not in output with config_context(display='diagram'): output = tree._repr_mimebundle_() assert "text/plain" in output assert "text/html" in output