From 01e6d0b44d4fbc43bcbb1fe8965cb1a42fcefe75 Mon Sep 17 00:00:00 2001
From: mdelitroz <maxime.delitroz@idiap.ch>
Date: Wed, 19 Jul 2023 14:29:50 +0200
Subject: [PATCH] updated tests for Montgomery dataset
---
tests/test_mc.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/test_mc.py b/tests/test_mc.py
index 87de46ea..ec4d85cd 100644
--- a/tests/test_mc.py
+++ b/tests/test_mc.py
@@ -15,7 +15,7 @@ def test_protocol_consistency():
datamodule = importlib.import_module(
"ptbench.data.montgomery.default"
).datamodule
- subset = datamodule.dataset_split.subsets
+ subset = datamodule.database_split.subsets
assert len(subset) == 3
@@ -126,7 +126,7 @@ def test_loading():
assert isinstance(data, torch.Tensor)
- assert data.size in (
+ assert data.size() in (
(1, 4020, 4892), # portrait
(1, 4892, 4020), # landscape
(1, 512, 512), # test database @ CI
@@ -143,7 +143,7 @@ def test_loading():
datamodule = importlib.import_module(
"ptbench.data.montgomery.default"
).datamodule
- subset = datamodule.database_split.subsetss
+ subset = datamodule.database_split.subsets
raw_data_loader = datamodule.raw_data_loader
# Need to use private function so we can limit the number of samples to use
--
GitLab