def testcase1(self): rootFolder = Directory() folderNames = '/home/gcj/finals' rootFolder.addDirectory(folderNames.split('/')[1:]) self.assertEqual(3, rootFolder.size()) folderNames = '/home/gcj/quals' rootFolder.addDirectory(folderNames.split('/')[1:]) self.assertEqual(4, rootFolder.size())