chore: Update deprecated functions (#398)

Signed-off-by: Wonyeong Choi <ciwnyg0815@gmail.com>
amundsen-io · Nov 2, 2020 · 8a559f5 · 8a559f5
1 parent e8679aa
commit 8a559f5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/unit/extractor/test_deltalake_extractor.py b/tests/unit/extractor/test_deltalake_extractor.py
@@ -219,7 +219,7 @@ def test_scrape_all_tables(self) -> None:
                   Table(name="test_table3", database="test_schema1", description=None,
                         tableType="delta", isTemporary=False)]
         actual = self.dExtractor.scrape_all_tables(tables)
-        self.assertEquals(2, len(actual))
+        self.assertEqual(2, len(actual))
 
 
 if __name__ == '__main__':

diff --git a/tests/unit/models/test_badge.py b/tests/unit/models/test_badge.py
@@ -28,11 +28,11 @@ def setUp(self) -> None:
 
     def test_get_badge_key(self) -> None:
         badge_key = self.badge_metada.get_badge_key(badge1.name)
-        self.assertEquals(badge_key, badge1.name)
+        self.assertEqual(badge_key, badge1.name)
 
     def test_create_nodes(self) -> None:
         nodes = self.badge_metada.create_nodes()
-        self.assertEquals(len(nodes), 2)
+        self.assertEqual(len(nodes), 2)
 
         node1 = {
             NODE_KEY: BadgeMetadata.BADGE_KEY_FORMAT.format(badge=badge1.name),
@@ -75,7 +75,7 @@ def test_bad_entity_label(self) -> None:
 
     def test_create_relation(self) -> None:
         relations = self.badge_metada.create_relation()
-        self.assertEquals(len(relations), 2)
+        self.assertEqual(len(relations), 2)
 
         relation1 = {
             RELATION_START_LABEL: self.badge_metada.start_label,