def test_pretty_key(self): visitor = ShardingByContestVisitor(Mock()) self.assertEqual(visitor.pretty_key('1'), 'Contest #1')
def test_key(self): visitor = ShardingByContestVisitor(Mock()) self.assertEqual(visitor.build_key(Mock(problem_id=('1', '2'))), '1')