Skip to content
Snippets Groups Projects
Commit 2ce9f658 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[tests] Setup minimal testing for nih-cxr14 database

parent ec5e1166
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
Pipeline #76827 passed
...@@ -17,7 +17,8 @@ def id_function(val): ...@@ -17,7 +17,8 @@ def id_function(val):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"split,lenghts", "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 ids=id_function, # just changes how pytest prints it
) )
...@@ -27,7 +28,7 @@ def test_protocol_consistency( ...@@ -27,7 +28,7 @@ def test_protocol_consistency(
from ptbench.data.nih_cxr14.datamodule import make_split from ptbench.data.nih_cxr14.datamodule import make_split
database_checkers.check_split( database_checkers.check_split(
make_split(f"{split}.json.bz2"), make_split(split),
lengths=lenghts, lengths=lenghts,
prefixes=("images/000",), prefixes=("images/000",),
possible_labels=(0, 1), possible_labels=(0, 1),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment