diff --git a/tests/test_ident_map.py b/tests/test_ident_map.py index 37ecf44..2ddb3db 100644 --- a/tests/test_ident_map.py +++ b/tests/test_ident_map.py @@ -1,8 +1,8 @@ from tsstats import Clients, Client ident_map = { - 1: 2, - 5: 2, + '1': '2', + '5': '2', 'UID1': 'UID2', 'UID5': 'UID2' } @@ -10,8 +10,8 @@ clients = Clients(ident_map) def test_get_id(): - assert clients[1].identifier == 2 - assert clients[5].identifier == 2 + assert clients['1'].identifier == '2' + assert clients['5'].identifier == '2' def test_get_uid():