final update

StanislavSol · Jul 10, 2023 · a2dcf1c · a2dcf1c
1 parent a4c644b
commit a2dcf1c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/gendiff/formats/plain.py b/gendiff/formats/plain.py
@@ -1,10 +1,11 @@
 def get_formatted_value(value):
-    elem_types = {"<class 'NoneType'>": 'null',
-                  "<class 'bool'>": str(value).lower(),
-                  "<class 'str'>": f'\'{value}\'',
-                  "<class 'int'>": value,
-                  "<class 'dict'>": '[complex value]'}
-    type_value = str(type(value))
+    elem_types = {type(None): 'null',
+                  type(bool): str(value).lower(),
+                  type(''): f'\'{value}\'',
+                  type(0): value,
+                  type({}): '[complex value]'}
+
+    type_value = type(value)
     if type_value in elem_types:
         return elem_types[type_value]
     return value

diff --git a/tests/test_recursion_comparison.py b/tests/test_recursion_comparison.py
@@ -1,6 +1,5 @@
 from gendiff import generate_diff
 import pytest
-LAST_ELEM = -1
 
 
 @pytest.mark.parametrize('path1, path2, formatter, expected', [