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')],
),
})
