Correct tests.

Author: jasongrout

ipywidgets/widgets/tests/test_widget.py

@@ -3,10 +3,10 @@
 
 """Test Widget."""
 
+from IPython import version_info
 from IPython.core.interactiveshell import InteractiveShell
 from IPython.display import display
 from IPython.utils.capture import capture_output
-
 from ..widget import Widget
 
 
@@ -17,8 +17,13 @@ def test_no_widget_view():
 
     with capture_output() as cap:
         w = Widget()
+        msg = w._repr_mimebundle_()
         display(w)
 
-    assert cap.outputs == [], repr(cap.outputs)
+    if version_info[0] > 6 or (version_info[0] == 6 and version_info[1] >= 1):
+        assert len(cap.outputs) == 1 and cap.outputs[0].data == msg, repr(cap.outputs)
+    else:
+        assert cap.outputs == [], repr(cap.outputs)
+
     assert cap.stdout == '', repr(cap.stdout)
     assert cap.stderr == '', repr(cap.stderr)