Convert decimals to integers if possible

polypheny · May 16, 2024 · 8df2206 · 8df2206
1 parent a5909b6
commit 8df2206
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -1,3 +1,4 @@
+import decimal
 import sys
 
 import pytest
@@ -25,8 +26,13 @@ def add_cur(request, doctest_namespace):
     cur.close()
 
     def myprint(*objects, sep=' ', end='\n', file=None, flush=False):
+        def toint(i):
+            if isinstance(i, decimal.Decimal) and int(i) == i:
+                return int(i)
+            else:
+                return i
         if len(objects) == 1 and isinstance(objects[0], dict):
-            print('{' + ', '.join(map(lambda i: f'{repr(i[0])}: {repr(i[1])}', sorted(objects[0].items()))) + '}')
+            print('{' + ', '.join(map(lambda i: f'{repr(i[0])}: {repr(toint(i[1]))}', sorted(objects[0].items()))) + '}')
         else:
             print(*objects, sep=sep, end=end, file=file, flush=flush)