added tests for iter_file_dups

This commit is contained in:
Darko Poljak 2014-01-28 08:36:14 +01:00
parent 63bb9779dd
commit 3ddd76fcce

View file

@ -3,7 +3,7 @@
# License: GPLv3 # License: GPLv3
import unittest import unittest
from sweeper import file_dups from sweeper import file_dups, iter_file_dups
import os import os
mydir = os.path.dirname(os.path.realpath(__file__)) mydir = os.path.dirname(os.path.realpath(__file__))
@ -23,6 +23,15 @@ class TestSweeper(unittest.TestCase):
for h, flist in dups.items(): for h, flist in dups.items():
self.assertTrue(len(flist) == 1) 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__': if __name__ == '__main__':
unittest.main() unittest.main()