def test_change_priority_none(self): result = main._change_priority('[delorean]', 10) self.assertEqual('[delorean]\npriority=10', result)
def test_change_priority_none_muilti(self): data = "[repo1]\n[repo2]\n" expected = "[repo1]\n{0}\n[repo2]\n{0}\n".format("priority=10") result = main._change_priority(data, 10) self.assertEqual(expected, result)