test_question12.py 文件源码

def test_count_upper_and_lower(self):
        test_data = "AAaaBBbb123CCcccd"
        right_answer = "upper=6,lower=8\n"
        with io.StringIO() as buf, redirect_stdout(buf):
            self.test_solve.count_upper_and_lower(test_data)
            my_answer = buf.getvalue()
        self.assertEqual(right_answer, my_answer)

        for i in range(10 ** 2):
            test_data = str()
            length = random.randint(1, 10 ** 3)
            for j in range(length):
                test_data += random.choice(string.ascii_letters + string.digits)
            right_answer = self.count_upper_and_lower(test_data)

            with io.StringIO() as buf, redirect_stdout(buf):
                self.test_solve.count_upper_and_lower(test_data)
                my_answer = buf.getvalue()
            self.assertEqual(right_answer, my_answer)