From 2ce9f6581e3996d3bbc42eeb7878400c7f631947 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Sat, 5 Aug 2023 21:17:36 +0200
Subject: [PATCH] [tests] Setup minimal testing for nih-cxr14 database
---
tests/test_nih_cxr14.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/test_nih_cxr14.py b/tests/test_nih_cxr14.py
index 242ac7e9..0870c99a 100644
--- a/tests/test_nih_cxr14.py
+++ b/tests/test_nih_cxr14.py
@@ -17,7 +17,8 @@ def id_function(val):
@pytest.mark.parametrize(
"split,lenghts",
[
- ("default", dict(train=98637, validation=6350, test=4054)),
+ ("default.json.bz2", dict(train=98637, validation=6350, test=4054)),
+ ("cardiomegaly.json", dict(train=40, validation=40)),
],
ids=id_function, # just changes how pytest prints it
)
@@ -27,7 +28,7 @@ def test_protocol_consistency(
from ptbench.data.nih_cxr14.datamodule import make_split
database_checkers.check_split(
- make_split(f"{split}.json.bz2"),
+ make_split(split),
lengths=lenghts,
prefixes=("images/000",),
possible_labels=(0, 1),
--
GitLab