import unittest
from contextlib import redirect_stdout
import io
import importlib
import user_code
import ast

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

class TestUserCode(unittest.TestCase):
    def test_variables_declared(self):
        """Check if all required variables are declared."""
        variables = ['steps_taken', 'step_goal', 'calories_burned', 'calorie_goal', 'result_message']
        missing_vars = [var for var in variables if not hasattr(user_code, var)]
        
        _dynamic_test(
            self,
            not missing_vars,
            "All required variables are declared.",
            f"Missing variable(s): {', '.join(missing_vars)}."
        )

    def test_variables_types(self):
        """Check if variables have the correct data types."""
        expected_types = {
            'steps_taken': int,
            'step_goal': int,
            'calories_burned': int,
            'calorie_goal': int,
            'result_message': bool
        }
        incorrect_types = {
            var: type(getattr(user_code, var)) 
            for var in expected_types if not isinstance(getattr(user_code, var), expected_types[var])
        }

        _dynamic_test(
            self,
            not incorrect_types,
            "All variables have the correct data types.",
            f"Incorrect data types detected: {incorrect_types}."
        )

    def test_result_message_logic(self):
        """Check if `result_message` is set correctly based on conditions."""
        # Extract values from user_code
        steps_taken = user_code.steps_taken
        step_goal = user_code.step_goal
        calories_burned = user_code.calories_burned
        calorie_goal = user_code.calorie_goal

        # Expected logic using `or`
        expected_value = steps_taken >= step_goal or calories_burned >= calorie_goal

        _dynamic_test(
            self,
            user_code.result_message == expected_value,
            f"The `result_message` variable is correctly set to {expected_value}.",
            f"Incorrect value for `result_message`. Expected {expected_value}, but got {user_code.result_message}."
        )

    def test_if_statement_is_single_and_uses_or(self):
        with open("user_code.py", "r") as f:
            tree = ast.parse(f.read())

        # Count all top-level `if` statements
        if_statements = [node for node in ast.walk(tree) if isinstance(node, ast.If)]

        # Check if at least one if statement contains an `or` operator
        or_found = any(
            isinstance(cond, ast.BoolOp) and isinstance(cond.op, ast.Or)
            for if_node in if_statements
            for cond in ast.walk(if_node.test)
        )

        _dynamic_test(
            self,
            len(if_statements) == 1 and or_found,
            "The solution correctly uses a single `if` statement with the `or` operator.",
            "The solution must use exactly one `if` statement and include the `or` operator."
        )

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


test_main.py

Explore the nuances of the conditional if-else operator in Python through clear explanations and practical examples. Strengthen your understanding with hands-on tasks that reinforce each concept, and discover how to use if-elif-else statements effectively for more complex decision-making.

Discover the versatility of the conditional if operator and its ability to shape program outcomes. Delve into the world of logical operators, master their practical application, and learn how to prioritize them effectively to write clear and efficient code.

Learn how to make decisions based on conditions, simplify code using the else keyword, improving both efficiency and readability. Unlock the efficiency of one-liner decisions with the ternary operator!

Discover the most advanced form of conditional statements: if-elif-else. Get familiar with its syntax and learn how to apply this construction correctly to handle multiple decision-making scenarios in your programs.

Challenge: Simplifying Fitness Tracker Logic

Solution