import unittest

import inflect


class test(unittest.TestCase):
    def test_classical(self):
        p = inflect.engine()

        # DEFAULT...

        self.assertEqual(p.plural_noun('error', 0),   'errors',      msg="classical 'zero' not active")
        self.assertEqual(p.plural_noun('wildebeest'), 'wildebeests', msg="classical 'herd' not active")
        self.assertEqual(p.plural_noun('Sally'),      'Sallys',      msg="classical 'names' active")
        self.assertEqual(p.plural_noun('brother'),    'brothers',    msg="classical others not active")
        self.assertEqual(p.plural_noun('person'),     'people',      msg="classical 'persons' not active")
        self.assertEqual(p.plural_noun('formula'),    'formulas',    msg="classical 'ancient' not active")

        # CLASSICAL PLURALS ACTIVATED...

        p.classical(all=True)
        self.assertEqual(p.plural_noun('error', 0),   'error',       msg="classical 'zero' active")
        self.assertEqual(p.plural_noun('wildebeest'), 'wildebeest',  msg="classical 'herd' active")
        self.assertEqual(p.plural_noun('Sally'),      'Sallys',      msg="classical 'names' active")
        self.assertEqual(p.plural_noun('brother'),    'brethren',    msg="classical others active")
        self.assertEqual(p.plural_noun('person'),     'persons',     msg="classical 'persons' active")
        self.assertEqual(p.plural_noun('formula'),    'formulae',    msg="classical 'ancient' active")

        # CLASSICAL PLURALS DEACTIVATED...

        p.classical(all=False)
        self.assertEqual(p.plural_noun('error', 0),   'errors',      msg="classical 'zero' not active")
        self.assertEqual(p.plural_noun('wildebeest'), 'wildebeests', msg="classical 'herd' not active")
        self.assertEqual(p.plural_noun('Sally'),      'Sallies',     msg="classical 'names' not active")
        self.assertEqual(p.plural_noun('brother'),    'brothers',    msg="classical others not active")
        self.assertEqual(p.plural_noun('person'),     'people',      msg="classical 'persons' not active")
        self.assertEqual(p.plural_noun('formula'),    'formulas',    msg="classical 'ancient' not active")

        # CLASSICAL PLURALS REREREACTIVATED...

        p.classical()
        self.assertEqual(p.plural_noun('error', 0),   'error',       msg="classical 'zero' active")
        self.assertEqual(p.plural_noun('wildebeest'), 'wildebeest',  msg="classical 'herd' active")
        self.assertEqual(p.plural_noun('Sally'),      'Sallys',      msg="classical 'names' active")
        self.assertEqual(p.plural_noun('brother'),    'brethren',    msg="classical others active")
        self.assertEqual(p.plural_noun('person'),     'persons',     msg="classical 'persons' active")
        self.assertEqual(p.plural_noun('formula'),    'formulae',    msg="classical 'ancient' active")

if __name__ == '__main__':
    try:
        unittest.main()
    except SystemExit:
        pass