class ParseAIDTestCase(unittest.TestCase): def setUp(self): self.manager = AIDResultsManager({ 'foo': AIDResults( ['Title'], [(1, 'Madoka'), (2, 'Madoka'), (3, 'Madoka')], ), 'bar': AIDResults( ['Title'], [(4, 'Madoka'), (5, 'Madoka'), (6, 'Madoka')], ), }) def test_explicit_aid(self): self.assertEqual(self.manager.parse_aid('aid:12345', 'foo'), 12345) def test_explicit_key(self): self.assertEqual(self.manager.parse_aid('foo:2', 'foo'), 2) def test_implicit_key(self): self.assertEqual(self.manager.parse_aid('3', 'bar'), 6) def test_invalid_key_syntax(self): with self.assertRaises(InvalidSyntaxError): self.manager.parse_aid('#:3', 'foo') def test_invalid_key(self): with self.assertRaises(ResultKeyError): self.manager.parse_aid('spam:3', 'foo') def test_invalid_number(self): with self.assertRaises(ResultNumberError): self.manager.parse_aid('foo:5', 'foo')
def setUp(self): self.manager = AIDResultsManager({ 'foo': AIDResults( ['Title'], [(1, 'Madoka'), (2, 'Madoka'), (3, 'Madoka')], ), 'bar': AIDResults( ['Title'], [(4, 'Madoka'), (5, 'Madoka'), (6, 'Madoka')], ), })