Exemple #1
0
from python_test_case import PythonTestCase, run_tests
import unittest

# Handle capturing what the students print
from io import StringIO
from unittest.mock import patch


class HelloWorldTest(PythonTestCase):
    def test(self):
        """ Should print 'Hello, World!' """
        # The above docstring is used as the description
        # of the test to the student

        with patch("sys.stdout", new=StringIO()) as output:

            # Import the module here, as the code will print as soon
            # as we import it
            import attempt
            self.assertEqual(output.getvalue().strip(), "Hello, World!")


if __name__ == "__main__":
    run_tests(HelloWorldTest)
Exemple #2
0
    def test_W_end(self):
        """The last value for the W array is correct"""
        self.assertAlmostEqual(attempt.W[-1], 9.494852380803035)

    def test_Z_end(self):
        """The last value for the Z array is correct"""
        self.assertAlmostEqual(attempt.Z[-1], 41.47999849170943)

    def test_W_start(self):
        """The first value for the W array is correct"""
        self.assertAlmostEqual(attempt.W[0], 12)

    def test_Z_start(self):
        """The first value for the Z array is correct"""
        self.assertAlmostEqual(attempt.Z[0], 40)

    def test_lengths(self):
        """The arrays all have the correct lengths """
        self.assertEqual(size(attempt.Z), 201)
        self.assertEqual(size(attempt.W), 201)

    def test_maxIndex(self):
        """The maxIndex variables are correct"""
        self.assertEqual(attempt.maxIndexZ, 113)
        self.assertEqual(attempt.maxIndexW, 134)


# Run the unit tests
if __name__ == "__main__":
    run_tests(Tests)
Exemple #3
0
    def test_executive_defined(self):
        """ Tests that the Executive class is defined """
        self.assertClassDefined(attempt, "Executive")

    def test_executive_inheritance(self):
        """ Tests that the Executive inherits from Employee """
        self.assertIsSubclass(attempt.Executive, attempt.Employee)

    def test_executive_init(self):
        """ Tests that the Executive has a correct __init__ method """
        self.assertMethodDefined(attempt.Executive, "__init__", 4)

        executive = attempt.Executive("Joseph Bloggs", 25000, 10000)

    @patch("attempt.Employee.wage")
    def test_executive_wage(self, emp_wage):
        """ Test that the Executive's wage function is correct """

        emp_wage.return_value = 25000/26

        executive = attempt.Executive("Joseph Bloggs", 25000, 10000)

        expected_wage = 25000/26 + 10000/26

        self.assertEqual(executive.wage(), expected_wage)
        self.assertCalled(emp_wage, "super().wage()")

if __name__ == "__main__":
    run_tests(InheritanceTests)