Function test_task_manager_loading

scripts/test_task_value_integration.py:22–69

Test that TaskManager loads and applies task values correctly

()

def test_task_manager_loading():
    """Test that TaskManager loads and applies task values correctly"""
    print("="*60)
    print("TEST 1: TaskManager Task Value Loading")
    print("="*60)

    # Initialize TaskManager with task values
    task_manager = TaskManager(
        task_source_type="parquet",
        task_source_path="./gdpval",
        task_values_path="./scripts/task_value_estimates/task_values.jsonl",
        default_max_payment=50.0
    )

    # Load tasks
    num_tasks = task_manager.load_tasks()
    print(f"\n✅ Loaded {num_tasks} tasks")

    # Check task values loaded
    if task_manager.task_values:
        print(f"✅ Task values loaded: {len(task_manager.task_values)} entries")
        values = list(task_manager.task_values.values())
        print(f"   Price range: ${min(values):.2f} - ${max(values):.2f}")
        print(f"   Average: ${sum(values)/len(values):.2f}")
    else:
        print("❌ No task values loaded!")
        return False

    # Select a task and verify it has max_payment
    task = task_manager.select_daily_task(date="2025-01-20")

    if 'max_payment' in task:
        print(f"\n✅ Task has max_payment field: ${task['max_payment']:.2f}")
        print(f"   Task ID: {task['task_id']}")
        print(f"   Occupation: {task['occupation']}")

        # Verify it matches the loaded value
        expected = task_manager.task_values.get(task['task_id'], 50.0)
        if abs(task['max_payment'] - expected) < 0.01:
            print(f"✅ Payment matches expected value")
        else:
            print(f"❌ Payment mismatch: got ${task['max_payment']:.2f}, expected ${expected:.2f}")
            return False
    else:
        print("❌ Task missing max_payment field!")
        return False

    return True


def test_fallback_behavior():