Skip to content

Commit e2678d2

Browse files
committed
fix folderparser test
1 parent d0070ac commit e2678d2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/util/test_folderparser.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,13 +87,13 @@ def test_parse_classification_folder_structure(self):
8787
self.assertEqual(img["annotationfile"]["classification_label"], "no-corrosion")
8888

8989
def test_parse_multilabel_classification_csv(self):
90-
folder = f"{thisdir}/../datasets/multilabel-classification"
90+
folder = f"{thisdir}/../datasets/skinproblem-multilabel-classification"
9191
parsed = folderparser.parsefolder(folder, is_classification=True)
9292
images = {img["name"]: img for img in parsed["images"]}
93-
img1 = images.get("img1.jpg")
93+
img1 = images.get("101_jpg.rf.ffb91e580c891eb04b715545274b2469.jpg")
9494
self.assertIsNotNone(img1)
9595
self.assertEqual(img1["annotationfile"]["type"], "classification_multilabel")
96-
self.assertEqual(set(img1["annotationfile"]["labels"]), {"rust"})
96+
self.assertEqual(set(img1["annotationfile"]["labels"]), {"Blackheads"})
9797

9898

9999
def _assertJsonMatchesFile(actual, filename):

0 commit comments

Comments
 (0)