import unittest import io import re from unittest import mock import math import student_X as main #Replace student_X with your filename import sys import io from io import StringIO class PrintHypotenuseLengthTests(unittest.TestCase): def test_default_case(self): function = main.print_hypotenuse_length try: docstring = function.__doc__ docstring = docstring.strip() except: docstring = "" correct = docstring != "" message = "docstring is blank." self.assertTrue(correct, message) with mock.patch('sys.stdout', new=io.StringIO()) as fake_stdout: function(3,4) output = fake_stdout.getvalue().strip() correct = "5.0" feedback = "Doesn't print the correct number when the parameters are 3 and 4." self.assertTrue(output == correct, feedback) with mock.patch('sys.stdout', new=io.StringIO()) as fake_stdout: function(8,6) output = fake_stdout.getvalue().strip() correct = "10.0" feedback = "Doesn't print the correct number when the parameters are 8 and 6." self.assertTrue(output == correct, feedback) with mock.patch('sys.stdout', new=io.StringIO()) as fake_stdout: function(10,10) output = fake_stdout.getvalue().strip() correct = str(10 * math.sqrt(2)) feedback = "Doesn't print the correct number when the parameters are 10 and 10." self.assertTrue(output == correct, feedback) if __name__ == "__main__": tests = PrintHypotenuseLengthTests() tests.test_default_case()