import unittest import io import re from unittest import mock import math import student_X as main #Replace student_X with your filename class AreaOfTriangleTests(unittest.TestCase): def test_default_case(self): function = main.area_of_triangle try: docstring = function.__doc__ docstring = docstring.strip() except: docstring = "" correct = docstring != "" message = "No docstring." self.assertTrue(correct, message) test = False try: base = 5 height = 1 correct = 2.5 result = function(base, height) test = correct == result except: pass message = "Doesn't return the correct area for a small triangle." self.assertTrue(test, message) test = False try: base = 5 height = 100 correct = 250 result = function(base, height) test = correct == result except: pass message = "Doesn't return the correct area for a tall triangle." self.assertTrue(test, message) if __name__ == "__main__": tests = AreaOfTriangleTests() tests.test_default_case()