Fix version comparison handling.

Fixes Teemu#159
Natim · Nov 5, 2018 · 2212197 · 2212197
1 parent e8cb804
commit 2212197
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytest_sugar.py b/pytest_sugar.py
@@ -14,6 +14,7 @@
 import os
 import re
 import sys
+from packaging.version import parse
 
 try:
     from configparser import ConfigParser
@@ -398,7 +399,7 @@ def pytest_runtest_logstart(self, nodeid, location):
         # show the module_name & in verbose mode the test name.
         pass
 
-    if pytest.__version__ >= '3.4':
+    if parse(pytest.__version__) >= parse('3.4'):
 
         def pytest_runtest_logfinish(self):
             # prevent the default implementation to try to show