import unittest import io import re from unittest import mock import math import student_X as main #Replace student_X with your filename class OddCharactersOnlyTests(unittest.TestCase): def test_default_case(self): function = main.odd_characters_only test = False try: incorrect = "" result = function.__doc__.strip() test = incorrect != result except: pass message = "No docstring." self.assertTrue(test) inputs = ('cheese') correct = 'hee' if isinstance(inputs, tuple): result = function(*inputs) else: result = function(inputs) message = "Doesn't return the correct thing on 'cheese'." self.assertTrue(result == correct, message) inputs = ('bananarama in the park.') correct = 'aaaaai h ak' if isinstance(inputs, tuple): result = function(*inputs) else: result = function(inputs) message = "Doesn't work on a long string." self.assertTrue(result == correct, message) if __name__ == "__main__": tests = OddCharactersOnlyTests() tests.test_default_case()