def test_get_size(self):
f1 = os.path.join(self.temp_dir, 'f1')
f2 = os.path.join(self.temp_dir, 'f2')
d1 = os.path.join(self.temp_dir, 'tempfolder')
os.mkdir(d1)
f3 = os.path.join(d1, 'f3')
self._create_file(f1, 10)
self._create_file(f2, 10)
self._create_file(f3, 10)
self.assertEqual(get_size(f1), 10, 'One file created not correct size')
self.assertEqual(get_size(self.temp_dir), 30, 'Folder size not correct')
def get_size(self):
return get_size(self.path)
