Skip to content

Commit 7324861

Browse files
committed
removed test and comments
1 parent d5d9386 commit 7324861

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

test/test_prototype_builtin_datasets.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -190,17 +190,10 @@ def test_label_matches_path(self, test_home, dataset_mock, config):
190190
dataset = datasets.load(dataset_mock.name, **config)
191191

192192
for sample in dataset:
193-
# check if correct keys exist
194193
assert "image" in sample
195194
assert "label" in sample
196195

197-
# check if correct instance type
198196
assert isinstance(sample["image"], Image)
199197
assert isinstance(sample["label"], Label)
200198

201-
# check is data type is correct
202-
assert isinstance(sample["image"].data, torch.FloatTensor)
203-
assert isinstance(sample["label"].data, torch.LongTensor)
204-
205-
# verify image size is (1, 16, 16)
206199
assert sample["image"].shape == (1, 16, 16)

0 commit comments

Comments
 (0)