import unittest
import re

class TestSum(unittest.TestCase):
    def test_numbers_declared(self):
        import user_code
        self.assertTrue(hasattr(user_code, "numbers"), "Expected `numbers` to be declared.")
    
    def test_total_declared(self):
        import user_code
        self.assertTrue(hasattr(user_code, "total"), "Expected `total` to be declared.")
    
    def test_total_value_correct(self):
        import user_code
        expected = sum(user_code.numbers)
        actual = user_code.total
        self.assertEqual(actual, expected, f"Expected total to be {expected}, got {actual}.")
    
    def test_sum_function_used(self):
        with open("user_code.py") as f:
            src = f.read()
        self.assertRegex(src, r"sum\s*\(", "Expected the `sum()` function to be used.")
    
    def test_total_is_int(self):
        import user_code
        self.assertIsInstance(user_code.total, int, "Expected `total` to be an int.")

if __name__ == "__main__":
    unittest.main()


With broken test

Solution

Awesome!

With broken test

Solution

Awesome!