"references/segmentation/utils.py" did not exist on "8697f9e0eeaf9191a3204106bb08ccbe68c0739f"
Commit 6cc8179b authored by peterjc123's avatar peterjc123 Committed by Soumith Chintala
Browse files

Fix for folder tests on Windows (#952)

parent 249cfbf5
......@@ -14,11 +14,11 @@ def mock_transform(return_value, arg_list):
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']
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']]
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']]
def test_image_folder(self):
......@@ -51,7 +51,7 @@ class Tester(unittest.TestCase):
self.assertEqual(imgs, outputs)
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 = []
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