diff --git a/test/test_sweeper.py b/test/test_sweeper.py index c4aced3..d75797a 100644 --- a/test/test_sweeper.py +++ b/test/test_sweeper.py @@ -3,7 +3,7 @@ # License: GPLv3 import unittest -from sweeper import file_dups +from sweeper import file_dups, iter_file_dups import os mydir = os.path.dirname(os.path.realpath(__file__)) @@ -23,6 +23,15 @@ class TestSweeper(unittest.TestCase): for h, flist in dups.items(): self.assertTrue(len(flist) == 1) + def test_iter_fule_dups_rethash(self): + for item in iter_file_dups([os.path.join(mydir, 'testfiles_dups')], + rethash=True): + self.assertTrue(type(item).__name__ == 'tuple') + + def test_iter_fule_dups_norethash(self): + for item in iter_file_dups([os.path.join(mydir, 'testfiles_dups')]): + self.assertTrue(type(item).__name__ == 'list') + if __name__ == '__main__': unittest.main()