import unittest import io import re from unittest import mock import math import student_X as main #Replace student_X with your filename class HypotenuseTests(unittest.TestCase): def test_default_case(self): function = main.hypotenuse try: docstring = function.__doc__ docstring = docstring.strip() except: docstring = "" correct = docstring != "" message = "No docstring." self.assertTrue(correct, message) test = False try: correct = 5 result = function(3, 4) test = correct == result except: pass message = "Doesn't return the correct hypotenuse length for a (relatively) small right triangle" self.assertTrue(test, message) test = False try: correct = 100 * math.sqrt(2) result = function(100, 100) test = correct == result except: pass message = "Doesn't return the correct hypotenuse length for a bigger right triangle" self.assertTrue(test, message) if __name__ == "__main__": tests = HypotenuseTests() tests.test_default_case()