def test_get_project_base(self): config = alembic_config.Config() config.set_main_option('script_location', 'a.b.c:d') proj_base = cli._get_project_base(config) self.assertEqual('a', proj_base)
def mocked_root_dir(cfg): return os.path.join('/fake/dir', cli._get_project_base(cfg))