

@mock.patch('alembic.script.ScriptDirectory.walk_revisions')
def test__get_branch_points(self, walk_mock):
    revisions = [FakeRevision(is_branch_point=tools.get_random_boolean) for i in range(50)]
    walk_mock.return_value = revisions
    script_dir = alembic_script.ScriptDirectory.from_config(self.configs[0])
    self.assertEqual(set((rev for rev in revisions if rev.is_branch_point)), set(cli._get_branch_points(script_dir)))

dataset/ETHPy150Open openstack/neutron/neutron/tests/unit/db/test_migration.py TestCli.test__get_branch_points/original