def test_restrictCheckers(self): """ Test for method restrictCheckers. Manually set allowed messages, then check for the result of registered checkers after run this method. """ runner = Runner() runner.restrictCheckers(HeaderChecker.msgs.keys()[:1]) # After run it, only HeaderChecker should be left in # registered checkers registeredCheckers = sum(runner.linter._checkers.values(), []) self.assertEqual(len(registeredCheckers), 1) self.assertEqual(type(registeredCheckers[0]), HeaderChecker)
def test_restrictCheckers(self): """ Test for method restrictCheckers. Manually set allowed messages, then check for the result of registered checkers after run this method. """ runner = Runner() runner.restrictCheckers(list(HeaderChecker.msgs.keys())[:1]) # After run it, only HeaderChecker should be left in # registered checkers registeredCheckers = sum(list(runner.linter._checkers.values()), []) self.assertEqual(len(registeredCheckers), 1) self.assertEqual(type(registeredCheckers[0]), HeaderChecker)