"""Unit test for roman2.pyThis program is part of "Dive Into Python", a free Python book forexperienced programmers.  Visit http://diveintopython.org/ for thelatest version."""__author__ = "Mark Pilgrim (mark@diveintopython.org)"__version__ = "$Revision: 1.3 $"__date__ = "$Date: 2004/05/05 21:57:20 $"__copyright__ = "Copyright (c) 2001 Mark Pilgrim"__license__ = "Python"import roman2import unittestclass KnownValues(unittest.TestCase):    knownValues = ( (1, 'I'),                    (2, 'II'),                    (3, 'III'),                    (4, 'IV'),                    (5, 'V'),                    (6, 'VI'),                    (7, 'VII'),                    (8, 'VIII'),                    (9, 'IX'),                    (10, 'X'),                    (50, 'L'),                    (100, 'C'),                    (500, 'D'),                    (1000, 'M'),                    (31, 'XXXI'),                    (148, 'CXLVIII'),                    (294, 'CCXCIV'),                    (312, 'CCCXII'),                    (421, 'CDXXI'),                    (528, 'DXXVIII'),                    (621, 'DCXXI'),                    (782, 'DCCLXXXII'),                    (870, 'DCCCLXX'),                    (941, 'CMXLI'),                    (1043, 'MXLIII'),                    (1110, 'MCX'),                    (1226, 'MCCXXVI'),                    (1301, 'MCCCI'),                    (1485, 'MCDLXXXV'),                    (1509, 'MDIX'),                    (1607, 'MDCVII'),                    (1754, 'MDCCLIV'),                    (1832, 'MDCCCXXXII'),                    (1993, 'MCMXCIII'),                    (2074, 'MMLXXIV'),                    (2152, 'MMCLII'),                    (2212, 'MMCCXII'),                    (2343, 'MMCCCXLIII'),                    (2499, 'MMCDXCIX'),                    (2574, 'MMDLXXIV'),                    (2646, 'MMDCXLVI'),                    (2723, 'MMDCCXXIII'),                    (2892, 'MMDCCCXCII'),                    (2975, 'MMCMLXXV'),                    (3051, 'MMMLI'),                    (3185, 'MMMCLXXXV'),                    (3250, 'MMMCCL'),                    (3313, 'MMMCCCXIII'),                    (3408, 'MMMCDVIII'),                    (3501, 'MMMDI'),                    (3610, 'MMMDCX'),                    (3743, 'MMMDCCXLIII'),                    (3844, 'MMMDCCCXLIV'),                    (3888, 'MMMDCCCLXXXVIII'),                    (3940, 'MMMCMXL'),                    (3999, 'MMMCMXCIX'))    def testToRomanKnownValues(self):        """toRoman should give known result with known input"""        for integer, numeral in self.knownValues:            result = roman2.toRoman(integer)            self.assertEqual(numeral, result)    def testFromRomanKnownValues(self):        """fromRoman should give known result with known input"""        for integer, numeral in self.knownValues:            result = roman2.fromRoman(numeral)            self.assertEqual(integer, result)class ToRomanBadInput(unittest.TestCase):    def testTooLarge(self):        """toRoman should fail with large input"""        self.assertRaises(roman2.OutOfRangeError, roman2.toRoman, 4000)    def testZero(self):        """toRoman should fail with 0 input"""        self.assertRaises(roman2.OutOfRangeError, roman2.toRoman, 0)    def testNegative(self):        """toRoman should fail with negative input"""        self.assertRaises(roman2.OutOfRangeError, roman2.toRoman, -1)    def testNonInteger(self):        """toRoman should fail with non-integer input"""        self.assertRaises(roman2.NotIntegerError, roman2.toRoman, 0.5)class FromRomanBadInput(unittest.TestCase):    def testTooManyRepeatedNumerals(self):        """fromRoman should fail with too many repeated numerals"""        for s in ('MMMM', 'DD', 'CCCC', 'LL', 'XXXX', 'VV', 'IIII'):            self.assertRaises(roman2.InvalidRomanNumeralError, roman2.fromRoman, s)    def testRepeatedPairs(self):        """fromRoman should fail with repeated pairs of numerals"""        for s in ('CMCM', 'CDCD', 'XCXC', 'XLXL', 'IXIX', 'IVIV'):            self.assertRaises(roman2.InvalidRomanNumeralError, roman2.fromRoman, s)    def testMalformedAntecedent(self):        """fromRoman should fail with malformed antecedents"""        for s in ('IIMXCC', 'VX', 'DCM', 'CMM', 'IXIV',                  'MCMC', 'XCX', 'IVI', 'LM', 'LD', 'LC'):            self.assertRaises(roman2.InvalidRomanNumeralError, roman2.fromRoman, s)class SanityCheck(unittest.TestCase):    def testSanity(self):        """fromRoman(toRoman(n))==n for all n"""        for integer in range(1, 4000):            numeral = roman2.toRoman(integer)            result = roman2.fromRoman(numeral)            self.assertEqual(integer, result)class CaseCheck(unittest.TestCase):    def testToRomanCase(self):        """toRoman should always return uppercase"""        for integer in range(1, 4000):            numeral = roman2.toRoman(integer)            self.assertEqual(numeral, numeral.upper())    def testFromRomanCase(self):        """fromRoman should only accept uppercase input"""        for integer in range(1, 4000):            numeral = roman2.toRoman(integer)            roman2.fromRoman(numeral.upper())            self.assertRaises(roman2.InvalidRomanNumeralError,                              roman2.fromRoman, numeral.lower())if __name__ == "__main__":    unittest.main()