def test_length(self, mock_md):
loader = Mock()
extensions = None
transform = Mock()
target_transform = Mock()
images = ['a.1', 'b.1', 'c.1', 'd.1']
mock_md.return_value = images
dataset = ds.SimpleImageFolder('root', loader, extensions, transform, target_transform)
self.assertTrue(len(dataset) == len(images))
def test_transform_args(self, mock_md):
loader = Mock()
mock_image = Mock()
loader.return_value = mock_image
extensions = None
transform = Mock()
target_transform = Mock()
images = ['a.1', 'b.1', 'c.1', 'd.1']
mock_md.return_value = images
dataset = ds.SimpleImageFolder('root', loader, extensions, transform, target_transform)
_ = dataset[1]
self.assertTrue(transform.call_args[0][0] == mock_image)