def test_data_dir(self): project = Project(path=sample_project_dir) self.assertEqual(project.data_dir, '.crawlmi') project.set_data_dir('crawlmi_data') self.assertEqual(project.data_dir, join(sample_project_dir, 'crawlmi_data')) self.assertFalse(exists(project.data_dir)) project.data_path('.', create_dir=True) self.assertTrue(exists(project.data_dir)) os.rmdir(project.data_dir) project.set_data_dir(None) self.assertEqual(project.data_dir, '.crawlmi')
def test_data_path(self): project = Project(path=sample_project_dir) project.set_data_dir('crawlmi_data') # relative path expected = join(sample_project_dir, 'crawlmi_data', 'a', 'b') relative = project.data_path(join('a', 'b'), create_dir=False) self.assertEqual(relative, expected) self.assertFalse(exists(expected)) # create dir project.data_path(join('a', 'b'), create_dir=True) self.assertTrue(exists(expected)) os.rmdir(join(sample_project_dir, 'crawlmi_data', 'a', 'b')) os.rmdir(join(sample_project_dir, 'crawlmi_data', 'a')) # absolute path expected = os.path.abspath(__file__) absolute = project.data_path(expected, create_dir=False) self.assertEqual(absolute, expected) os.rmdir(project.data_dir)