From d799fd03315aacc9466e87682a3b5cbcab45a1b6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Ma=C5=9Blanka?= <piotr.maslanka@ericsson.com>
Date: Fri, 19 Apr 2024 08:06:53 +0200
Subject: [PATCH] fix unit tests

---
 tests/test_coding/test_structures.py | 2 +-
 tests/test_json.py                   | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_coding/test_structures.py b/tests/test_coding/test_structures.py
index f33ff62f..40526c13 100644
--- a/tests/test_coding/test_structures.py
+++ b/tests/test_coding/test_structures.py
@@ -15,7 +15,7 @@ from satella.coding.structures import TimeBasedHeap, Heap, typednamedtuple, \
     CacheDict, StrEqHashableMixin, ComparableIntEnum, HashableIntEnum, ComparableAndHashableBy, \
     ComparableAndHashableByInt, SparseMatrix, ExclusiveWritebackCache, Subqueue, \
     CountingDict, ComparableEnum, LRU, LRUCacheDict, Vector, DefaultDict, PushIterable, \
-    ComparableAndHashableByStr, NotEqualToAnything, NOT_EQUAL_TO_ANYTHING, DictionaryEQAble, SetZip, OnStrOnlyNameJSONAbleDataObject
+    ComparableAndHashableByStr, NotEqualToAnything, NOT_EQUAL_TO_ANYTHING, DictionaryEQAble, SetZip, OnStrOnlyName
 
 
 def continue_testing_omni(self, omni_class):
diff --git a/tests/test_json.py b/tests/test_json.py
index 2c71027d..334dffce 100644
--- a/tests/test_json.py
+++ b/tests/test_json.py
@@ -16,9 +16,9 @@ class TestJson(unittest.TestCase):
             language: str
 
         a = CultureContext(language='pl', timezone='Europe/Warsaw')
-        self.assertEquals(a.to_json(), {'language': 'pl', 'timezone': 'Europe/Warsaw', 'units': 'metric'})
-        self.assertEquals(CultureContext.from_json(a.to_json()), a)
-        self.assertEquals(hash(CultureContext.from_json(a.to_json())), hash(a))
+        self.assertEqual(a.to_json(), {'language': 'pl', 'timezone': 'Europe/Warsaw', 'units': 'metric'})
+        self.assertEqual(CultureContext.from_json(a.to_json()), a)
+        self.assertEqual(hash(CultureContext.from_json(a.to_json())), hash(a))
 
     def test_json_encoder_enums(self):
         enc = JSONEncoder()
-- 
GitLab