From 05941d9c7d1206d2facb61505a2aff7136d66705 Mon Sep 17 00:00:00 2001 From: Lukas Heumos Date: Fri, 15 Nov 2024 14:19:21 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Clean=20up=20test=20properly=20(#16?= =?UTF-8?q?1)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zethson --- tests/test_ontology.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/tests/test_ontology.py b/tests/test_ontology.py index b9d6500..10e988e 100644 --- a/tests/test_ontology.py +++ b/tests/test_ontology.py @@ -1,11 +1,20 @@ +from pathlib import Path + import bionty.base as bt_base from bionty.base.dev._io import s3_bionty_assets def test_ontology(): localpath = s3_bionty_assets("ontology_all__pw__7.79__Pathway") - onto = bt_base.Ontology(localpath) - assert onto.get_term("PW:0000014").name == "neurodegenerative pathway" - df = onto.to_df(source="pw", include_id_prefixes={"pw": ["PW"]}) - assert df.shape == (2647, 4) - assert df.index.name == "ontology_id" + + try: + onto = bt_base.Ontology(localpath) + assert onto.get_term("PW:0000014").name == "neurodegenerative pathway" + + df = onto.to_df(source="pw", include_id_prefixes={"pw": ["PW"]}) + assert df.shape == (2647, 4) + assert df.index.name == "ontology_id" + + finally: + if Path(localpath).exists: + Path(localpath).unlink()