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()
def testDirCount(self):
d = RandomTree()
dc = d.dirsCalculation(depth=1, dirsPerDir=0 )
#print "dirs x: %s" % d.dirs()
self.assertEqual( 1, dc )
dc = d.dirsCalculation(depth=1, dirsPerDir=1 )
self.assertEqual( 2, dc )
d.rm()
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()
