Use LooseVersion to work with e.g. 1.6.dev

sympy · Mar 31, 2020 · 3be8445 · 3be8445
1 parent 7338b07
commit 3be8445
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/benchmarks/polygon.py b/benchmarks/polygon.py
@@ -1,5 +1,5 @@
 from __future__ import print_function, division
-from distutils.version import LooseVersion, StrictVersion
+from distutils.version import LooseVersion
 from sympy import Rational
 import sympy
 from sympy.geometry import (Line, Point,
@@ -55,39 +55,39 @@ def time_bench01(self):
 
 class PolygonSecondMoment:
     def setup(self):
-        if StrictVersion(sympy.__version__) < StrictVersion("1.3.0"):
+        if LooseVersion(sympy.__version__) < LooseVersion("1.3.0"):
             raise NotImplementedError
     def time_bench01(self):
         "Polygon.second_moment_of_area() for w in range(10)"
         [listOfPolygons[w].second_moment_of_area() for w in range(10)]
 
 class PolygonFirstMoment:
     def setup(self):
-        if StrictVersion(sympy.__version__) < StrictVersion("1.5.0"):
+        if LooseVersion(sympy.__version__) < LooseVersion("1.5.0"):
             raise NotImplementedError
     def time_bench01(self):
         "Polygon.first_moment_of_area for w in range(10)"
         [listOfPolygons[w].first_moment_of_area() for w in range(10)]
 
 class PolygonPolarSecondMoment:
     def setup(self):
-        if StrictVersion(sympy.__version__) < StrictVersion("1.5.0"):
+        if LooseVersion(sympy.__version__) < LooseVersion("1.5.0"):
             raise NotImplementedError
     def time_bench01(self):
         "Polygon.polar_second_moment_of_area for w in range(10)"
         [listOfPolygons[w].polar_second_moment_of_area() for w in range(10)]
 
 class PolygonSectionModulus:
     def setup(self):
-        if StrictVersion(sympy.__version__) < StrictVersion("1.5.0"):
+        if LooseVersion(sympy.__version__) < LooseVersion("1.5.0"):
             raise NotImplementedError
     def time_bench01(self):
         "Polygon.section_modulus for w in range(10)"
         [listOfPolygons[w].section_modulus() for w in range(10)]
 
 class PolygonCutSection:
     def setup(self):
-        if StrictVersion(sympy.__version__) < StrictVersion("1.5.0"):
+        if LooseVersion(sympy.__version__) < LooseVersion("1.5.0"):
             raise NotImplementedError
     def time_bench01(self):
         "Polygon.cut_section for w in range(10)"