From ef0579b92267a888b15f28ea2ab17f553cf592c0 Mon Sep 17 00:00:00 2001 From: Vedanth Ramji <86137377+Vedanth-Ramji@users.noreply.github.com> Date: Wed, 15 Nov 2023 14:58:21 +0530 Subject: [PATCH] Fixed discrepancies in typing for deeparg tests --- tests/test_smoke.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_smoke.py b/tests/test_smoke.py index 09641c6..6da1f58 100644 --- a/tests/test_smoke.py +++ b/tests/test_smoke.py @@ -44,8 +44,8 @@ def test_add_aro_column_deeparg(is_hamronized, uses_manual_curation): assert set(normed.loc[normed['gene_symbol'] == 'MGRB', 'ARO'].tolist()) == {'ARO:mgrB'} else: assert normed.set_index('#ARG').loc['YKKD', 'ARO'] == 'ARO:3003064' - assert normed.set_index('#ARG').loc['PENA', 'ARO'] == 'ARO:3004832' - assert normed.set_index('#ARG').loc['PBP-2X', 'ARO'] == 'ARO:PBP-2X' + assert list(normed.set_index('#ARG').loc['PENA', 'ARO']) == ['ARO:3004832', 'ARO:3004832'] + assert list(normed.set_index('#ARG').loc['PBP-2X', 'ARO']) == ['ARO:PBP-2X', 'ARO:PBP-2X'] assert normed.set_index('#ARG').loc['TETR', 'ARO'] == 'ARO:3003479' assert normed.set_index('#ARG').loc['PATA', 'ARO'] == 'ARO:3000024'