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 PrintAverageTests(unittest.TestCase):
    
    def test_default_case(self):
        function = main.print_average
        

        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(7.5, 7.5)
        output = fake_stdout.getvalue().strip()

        correct = "7.5"
        feedback = "Doesn't print the correct number when the parameters are both 7.5"
        self.assertTrue(output == correct, feedback)


        
        with mock.patch('sys.stdout', new=io.StringIO()) as fake_stdout:
            function(2, -2)
        output = fake_stdout.getvalue().strip()

        correct = "0.0"
        feedback = "Doesn't print the correct number when the average is 0.0"
        self.assertTrue(output == correct, feedback)


        
        with mock.patch('sys.stdout', new=io.StringIO()) as fake_stdout:
            function(5, 14)
        output = fake_stdout.getvalue().strip()

        correct = "9.5"
        feedback = "Doesn't print the correct number when the parameters are 5 and 14."
        self.assertTrue(output == correct, feedback)
        
        
if __name__ == "__main__":
    tests = PrintAverageTests()
    tests.test_default_case()