def test_image_discovery_2():
"""
Test CSV Filename
"""
# Test just the CSV file name
image_discovery(path='./', filename='test')
assert os.path.isfile('test.csv')
os.remove('test.csv')
def test_image_discovery_3(mocked_glob, mocked_getsize):
"""
Test CSV Filesizes
"""
expected_values = pd.DataFrame(columns=['Filename', 'Size'],
data=[['test1.tif', 1], ['test2.tif', 4]])
# Test just the CSV file contents
image_discovery(path='./', filesize=True)
test = pd.read_csv('list.csv')
print test
assert test.equals(expected_values)
os.remove('list.csv')
def test_image_discovery_1(mocked_glob):
"""
Test simple CSV contents
"""
expected_values = pd.DataFrame(columns=['Filename'],
data=['test1.tif', 'test2.tif'])
# Test just the CSV file contents
image_discovery(path='./')
test = pd.read_csv('list.csv')
assert test.equals(expected_values)
os.remove('list.csv')