From 45043ddb2cbdd9793290cf4e0988842fd692ecd2 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Thu, 1 Feb 2024 10:56:43 +0100
Subject: [PATCH] [test] Remove unnecessary padchest test code
---
tests/test_padchest.py | 13 +------------
1 file changed, 1 insertion(+), 12 deletions(-)
diff --git a/tests/test_padchest.py b/tests/test_padchest.py
index 5fc342e1..98eaeb2d 100644
--- a/tests/test_padchest.py
+++ b/tests/test_padchest.py
@@ -60,18 +60,7 @@ def test_loading(database_checkers, name: str, dataset: str, num_labels: int):
datamodule.setup("predict") # sets up all datasets
if dataset in datamodule.predict_dataloader():
- # Not all datamodules have a test set
- try:
- loader = datamodule.predict_dataloader()[dataset]
- except KeyError as e:
- if str(e) == "test" and name in [
- "idiap",
- "no_tb_idiap",
- "cardiomegaly_idiap",
- ]:
- assert True
- else:
- assert False
+ loader = datamodule.predict_dataloader()[dataset]
limit = 3 # limit load checking
for batch in loader:
--
GitLab