From fcccc84bf8d90d115af600ff774033550de4c60e Mon Sep 17 00:00:00 2001 From: Sam Hughes Date: Tue, 29 Aug 2023 15:41:06 +0100 Subject: [PATCH] chore: reenable test --- tests/endpoints/test_vessels_real.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/endpoints/test_vessels_real.py b/tests/endpoints/test_vessels_real.py index e0bf43e6..a6497ea8 100644 --- a/tests/endpoints/test_vessels_real.py +++ b/tests/endpoints/test_vessels_real.py @@ -1,5 +1,6 @@ from tests.testcases import TestCaseUsingRealAPI from tests.timer import Timer +from vortexasdk import Products from vortexasdk.endpoints.vessels import Vessels @@ -45,15 +46,14 @@ def test_search_ids_dataframe(self): assert list(df.columns) == ["id", "name", "imo", "vessel_class"] assert len(df) == 2 - # TODO: This test fails. Reason is unknown. Removing for now. - # def test_find_crude_vessels(self): - # crude = [ - # p.id - # for p in Products().search("crude").to_list() - # if "group" in p.layer - # ] - # df = Vessels().search(vessel_product_types=crude).to_df() - # assert len(df) > 1000 + def test_find_crude_vessels(self): + crude = [ + p.id + for p in Products().search("crude").to_list() + if "group" in p.layer + ] + df = Vessels().search(vessel_product_types=crude).to_df() + assert len(df) > 1000 def test_load_all(self): all_products = Vessels().load_all()