def test_makedirs(self):
new_tmp_dir = "testmakedirs/"
nested_dir = new_tmp_dir + "test_nest_dir"
pfio.makedirs("file://" + nested_dir)
self.assertTrue(os.path.isdir(nested_dir))
pfio.remove(new_tmp_dir, True)
def test_remove(self):
test_file = "test_remove.txt"
test_dir = "test_dir/"
nested_dir = os.path.join(test_dir, "nested_file/")
nested_file = os.path.join(nested_dir, test_file)
with pfio.open(test_file, 'w') as fp:
fp.write('foobar')
# test remove on one file
self.assertTrue(pfio.exists(test_file))
pfio.remove(test_file)
self.assertFalse(pfio.exists(test_file))
# test remove on directory
pfio.makedirs(nested_dir)
with pfio.open(nested_file, 'w') as fp:
fp.write('foobar')
self.assertTrue(pfio.exists(test_dir))
self.assertTrue(pfio.exists(nested_dir))
self.assertTrue(pfio.exists(nested_file))
pfio.remove(test_dir, True)
self.assertFalse(pfio.exists(test_dir))
self.assertFalse(pfio.exists(nested_dir))
self.assertFalse(pfio.exists(nested_file))
def test_list(self):
nested_dir_name1 = "nested_dir1"
nested_dir_name2 = "nested_dir2"
nested_dir_path1 = os.path.join(self.tmpdir.name, nested_dir_name1)
nested_dir_path2 = os.path.join(nested_dir_path1, nested_dir_name2)
nested_dir_path2_relative = os.path.join(nested_dir_name1,
nested_dir_name2)
pfio.makedirs(nested_dir_path1)
pfio.makedirs(nested_dir_path2)
file_list = list(pfio.list(self.tmpdir.name))
self.assertIn(nested_dir_name1, file_list)
self.assertIn(self.tmpfile_name, file_list)
self.assertNotIn(nested_dir_path2_relative, file_list)
file_list = list(pfio.list(self.tmpdir.name, recursive=True))
self.assertIn(nested_dir_name1, file_list)
self.assertIn(self.tmpfile_name, file_list)
self.assertIn(nested_dir_path2_relative, file_list)