37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
#!/usr/bin/env python
|
|
# Author: Darko Poljak <darko.poljak@gmail.com>
|
|
# License: GPLv3
|
|
|
|
import unittest
|
|
from sweeper import file_dups, iter_file_dups
|
|
import os
|
|
|
|
mydir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
class TestSweeper(unittest.TestCase):
|
|
def test_file_dups_dups(self):
|
|
dups = file_dups([os.path.join(mydir, 'testfiles_dups')], 'md5')
|
|
dups_exist = False
|
|
for h, flist in dups.items():
|
|
if len(flist) > 1:
|
|
dups_exist = True
|
|
self.assertTrue(dups_exist)
|
|
|
|
def test_file_dups_nodups(self):
|
|
dups = file_dups([os.path.join(mydir, 'testfiles_nodups')], 'md5')
|
|
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()
|