Smarter version compare.

Author: jasongrout

ipywidgets/widgets/tests/test_widget.py

@@ -20,7 +20,7 @@ def test_no_widget_view():
         msg = w._repr_mimebundle_()
         display(w)
 
-    if version_info[0] > 6 or (version_info[0] == 6 and version_info[1] >= 1):
+    if version_info >= (6, 1):
         assert len(cap.outputs) == 1 and cap.outputs[0].data == msg, repr(cap.outputs)
     else:
         assert cap.outputs == [], repr(cap.outputs)

ipywidgets/widgets/widget.py

@@ -730,8 +730,7 @@ def _ipython_display_(self, **kwargs):
         Note: if we are in IPython 6.1 or later, we return NotImplemented so
         that _repr_mimebundle_ is used directly.
         """
-        if (ipython_version_info[0] > 6
-            or (ipython_version_info[0] == 6 and ipython_version_info[1] >= 1)):
+        if ipython_version_info >= (6, 1):
             raise NotImplementedError
 
         data = self._repr_mimebundle_(**kwargs)