def test_pretty_key(self):
     visitor = ShardingByScoringVisitor(Mock())
     self.assertEqual(visitor.pretty_key('1'), 'Scoring type: 1')
 def test_key(self):
     visitor = ShardingByScoringVisitor(Mock())
     self.assertEqual(visitor.build_key(Mock(scoring='ACM')), 'ACM')
     self.assertEqual(visitor.build_key(Mock(scoring='not_acm')), 'kirov')