Skip to content

GitLab

Commit 6cc8179b authored by peterjc123's avatar peterjc123 Committed by Soumith Chintala
Browse files

Fix for folder tests on Windows (#952)

parent 249cfbf5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/test_folder.py
View file @ 6cc8179b
...@@ -14,11 +14,11 @@ def mock_transform(return_value, arg_list): ...@@ -14,11 +14,11 @@ def mock_transform(return_value, arg_list):
class Tester(unittest.TestCase): class Tester(unittest.TestCase):
root = get_file_path_2('test/assets/dataset/') root = os.path.normpath(get_file_path_2('test/assets/dataset/'))
classes = ['a', 'b'] classes = ['a', 'b']
class_a_images = [get_file_path_2(os.path.join('test/assets/dataset/a/', path)) class_a_images = [os.path.normpath(get_file_path_2(os.path.join('test/assets/dataset/a/', path)))
for path in ['a1.png', 'a2.png', 'a3.png']] for path in ['a1.png', 'a2.png', 'a3.png']]
class_b_images = [get_file_path_2(os.path.join('test/assets/dataset/b/', path)) class_b_images = [os.path.normpath(get_file_path_2(os.path.join('test/assets/dataset/b/', path)))
for path in ['b1.png', 'b2.png', 'b3.png', 'b4.png']] for path in ['b1.png', 'b2.png', 'b3.png', 'b4.png']]
def test_image_folder(self): def test_image_folder(self):
...@@ -51,7 +51,7 @@ class Tester(unittest.TestCase): ...@@ -51,7 +51,7 @@ class Tester(unittest.TestCase):
self.assertEqual(imgs, outputs) self.assertEqual(imgs, outputs)
def test_transform(self): def test_transform(self):
return_value = get_file_path_2('test/assets/dataset/a/a1.png') return_value = os.path.normpath(get_file_path_2('test/assets/dataset/a/a1.png'))
args = [] args = []
transform = mock_transform(return_value, args) transform = mock_transform(return_value, args)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment