예제 #1
0
 def setUp(self):
     self.thecube = Thecube()
예제 #2
0
class ThecubeTests(unittest.TestCase):

    def setUp(self):
        self.thecube = Thecube()

    def testIsPrimeNumber(self):
        self.assertTrue(self.thecube.isPrimeNumber(1))
        self.assertTrue(self.thecube.isPrimeNumber(2))
        self.assertTrue(self.thecube.isPrimeNumber(5))
        self.assertTrue(self.thecube.isPrimeNumber(7))
        self.assertTrue(self.thecube.isPrimeNumber(11))
        self.assertTrue(self.thecube.isPrimeNumber(13))
        self.assertTrue(self.thecube.isPrimeNumber(17))
        self.assertTrue(self.thecube.isPrimeNumber(19))
        self.assertTrue(self.thecube.isPrimeNumber(23))
        self.assertTrue(self.thecube.isPrimeNumber(29))
        self.assertTrue(self.thecube.isPrimeNumber(31))

        self.assertFalse(self.thecube.isPrimeNumber(4))
        self.assertFalse(self.thecube.isPrimeNumber(6))
        self.assertFalse(self.thecube.isPrimeNumber(8))
        self.assertFalse(self.thecube.isPrimeNumber(9))

    def testAddDigits(self):
        self.assertEquals(3, self.thecube.addDigits('012'))
        self.assertEquals(6, self.thecube.addDigits('222'))
        self.assertEquals(45, self.thecube.addDigits('0123456789'))
        self.assertEquals(3, self.thecube.addDigits('1hofei1lf00f1kd'))

    def testContainsPrimeNumber(self):
        self.assertTrue(self.thecube.containsPrimeNumber('1', '1', '1'))
        self.assertTrue(self.thecube.containsPrimeNumber('122', '1', '1'))
        self.assertTrue(self.thecube.containsPrimeNumber('28733', '13412', '96476'))
        self.assertFalse(self.thecube.containsPrimeNumber('22', '22', '22'))