예제 #1
0
 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)))
예제 #2
0
 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)))