def depthFunction(self, depth, dirsPerDir, filesPerDir ): d = RandomTree( depth=depth, dirsPerDir=dirsPerDir, filesPerDir=filesPerDir ) dirsCalc = d.dirsCalculation(depth, dirsPerDir) self.assertEqual( dirsCalc, d.dirsCount()) self.assertEqual( dirsCalc * filesPerDir, d.filesCount()) d.rm()
def testWalk(self): d = RandomTree( depth = 2, dirsPerDir = 2, filesPerDir = 4 ) root =d.rootDir() ( dirCount, fileCount) = self._w.count( root ) self.assertEqual( dirCount, d.dirsCount()) self.assertEqual( fileCount, d.filesCount()) self.assertEqual( sorted( [ x for x in d.dirs() ] ), sorted( d.dirs())) self.assertEqual( sorted( [ x for x in d.files() ]), sorted( d.files())) d.rm()