import unittest

class myTestSuite(unittest.TestCase):
    def setup(self):
        pass

    def tearDown(self):
        pass

    def number2roman(self, number):
	if(number == 2):
	    return "II"
        return "I" 

    def test_1_return_I(self):
        self.assertEqual(self.number2roman(1), "I")

    def test_2_return_II(self):
        self.assertEqual(self.number2roman(2), "II")
if __name__ == "__main__":
    unittest.main()

@@ -157,16 +157,52 @@
 umber):%0A
+%09if(number == 3):%0A%09    return %22III%22%0A
 %09if(numb
@@ -417,16 +417,104 @@
 ), %22II%22)
+%0A%0A    def test_3_return_III(self):%0A        self.assertEqual(self.number2roman(3), %22III%22)
 %0Aif __na

finish 3 to III