Fix precision test

shanefontaine · Dec 29, 2018 · 806efdf · 806efdf
1 parent a949b52
commit 806efdf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_public_client.py b/tests/test_public_client.py
@@ -79,8 +79,9 @@ def test_get_books(self, client, pair, precision, length):
         # Check Precision
         price = str(r[0][0])
         if precision == 'P0':
-            digits = len(price.split(".")[1])
-            assert digits == 1
+            digits = len(price.split("."))
+            # Will return either a whole number or a single decimal
+            assert (digits == 1 or digits == 2)
         elif precision == 'P1':
             assert len(price) == 4
         elif precision == 'P2':