def test_size(self):
""" Find files based on size criteria. """
# all files except the image files are less than 10 bytes
p_filter = SizeFilter(max_bytes=0)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(11, len(paths))
# only the image files contain data
p_filter = SizeFilter(min_bytes=1)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(6, len(paths))
# three images between 450 bytes and 9000
p_filter = SizeFilter(min_bytes=450, max_bytes=9000)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(3, len(paths))
def test_image_dimension(self):
""" Find images based on dimensions. """
p_filter = ImageDimensionFilter(max_width=1000, max_height=1000,
min_height=20, min_width=20)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(6, len(paths))
# ignore the 24x24
p_filter = ImageDimensionFilter(max_width=1000, max_height=1000,
min_height=25, min_width=25)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(5, len(paths))
# no 24x24, but only check it based on height
p_filter = ImageDimensionFilter(min_height=25)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(5, len(paths))
# only the 24x24
p_filter = ImageDimensionFilter(max_width=24, max_height=24)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(1, len(paths))
# only the 24x24, but only check it based on height
p_filter = ImageDimensionFilter(max_height=24)
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(1, len(paths))
# no parameters - all images
p_filter = ImageDimensionFilter()
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(6, len(paths))
def find(self, filepath):
from pathfinder import walk_and_filter
return walk_and_filter(filepath, self)
def test_color_image(self):
""" Find all color images. """
p_filter = ColorImageFilter()
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(2, len(paths))
def test_bw_image(self):
""" Find all grey scale images. """
p_filter = GreyscaleImageFilter()
paths = walk_and_filter(BASEPATH, p_filter)
self.assertEqual(4, len(paths))
def test_image(self):
""" Find all images. """
filter = ImageFilter()
paths = walk_and_filter(BASEPATH, filter)
self.assertEqual(6, len(paths))
