Skip to content

Commit 8a559f5

Browse files
authored
chore: Update deprecated functions (#398)
Signed-off-by: Wonyeong Choi <ciwnyg0815@gmail.com>
1 parent e8679aa commit 8a559f5

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/unit/extractor/test_deltalake_extractor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def test_scrape_all_tables(self) -> None:
219219
Table(name="test_table3", database="test_schema1", description=None,
220220
tableType="delta", isTemporary=False)]
221221
actual = self.dExtractor.scrape_all_tables(tables)
222-
self.assertEquals(2, len(actual))
222+
self.assertEqual(2, len(actual))
223223

224224

225225
if __name__ == '__main__':

tests/unit/models/test_badge.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@ def setUp(self) -> None:
2828

2929
def test_get_badge_key(self) -> None:
3030
badge_key = self.badge_metada.get_badge_key(badge1.name)
31-
self.assertEquals(badge_key, badge1.name)
31+
self.assertEqual(badge_key, badge1.name)
3232

3333
def test_create_nodes(self) -> None:
3434
nodes = self.badge_metada.create_nodes()
35-
self.assertEquals(len(nodes), 2)
35+
self.assertEqual(len(nodes), 2)
3636

3737
node1 = {
3838
NODE_KEY: BadgeMetadata.BADGE_KEY_FORMAT.format(badge=badge1.name),
@@ -75,7 +75,7 @@ def test_bad_entity_label(self) -> None:
7575

7676
def test_create_relation(self) -> None:
7777
relations = self.badge_metada.create_relation()
78-
self.assertEquals(len(relations), 2)
78+
self.assertEqual(len(relations), 2)
7979

8080
relation1 = {
8181
RELATION_START_LABEL: self.badge_metada.start_label,

0 commit comments

Comments
 (0)