checkpoint before simplifying data

test_new_format.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+"""
+Test the new format where AI returns a list and keys are generated locally.
+"""
+
+import json
+import sys
+import os
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from generate_prompts import JournalPromptGenerator
+
+def test_new_format():
+    """Test the new format where AI returns a list and keys are generated locally."""
+
+    print("Testing new format: AI returns list, keys generated locally")
+    print("="*60)
+
+    # Create a mock generator
+    generator = JournalPromptGenerator(config_path=".env")
+
+    # Create a mock AI response in the new list format
+    mock_ai_response = [
+        "Write about a childhood memory that still makes you smile.",
+        "Describe your perfect day from start to finish.",
+        "What is something you've been putting off and why?",
+        "Imagine you could have a conversation with any historical figure.",
+        "Write a letter to your future self one year from now.",
+        "Describe a place that feels like home to you."
+    ]
+
+    # Convert to JSON string
+    json_response = json.dumps(mock_ai_response)
+
+    print("\n1. Testing _parse_ai_response with list format:")
+    result = generator._parse_ai_response(json_response)
+    print(f"   Result type: {type(result)}")
+    print(f"   Number of prompts: {len(result)}")
+    print(f"   First prompt: {result[0][:50]}...")
+
+    # Verify it's a list of strings
+    assert isinstance(result, list), "Result should be a list"
+    assert all(isinstance(prompt, str) for prompt in result), "All items should be strings"
+
+    print("\n2. Testing add_prompts_to_pool with list of strings:")
+
+    # Get initial pool size
+    initial_pool_size = len(generator.pool_prompts)
+    print(f"   Initial pool size: {initial_pool_size}")
+
+    # Add prompts to pool
+    generator.add_prompts_to_pool(result)
+
+    # Check new pool size
+    new_pool_size = len(generator.pool_prompts)
+    print(f"   New pool size: {new_pool_size}")
+    print(f"   Added {new_pool_size - initial_pool_size} prompts")
+
+    # Check that prompts in pool have keys
+    print(f"\n3. Checking that prompts in pool have generated keys:")
+    for i, prompt_dict in enumerate(generator.pool_prompts[-len(result):]):
+        prompt_key = list(prompt_dict.keys())[0]
+        prompt_text = prompt_dict[prompt_key]
+        print(f"   Prompt {i+1}: Key='{prompt_key}', Text='{prompt_text[:30]}...'")
+        assert prompt_key.startswith("poolprompt"), f"Key should start with 'poolprompt', got '{prompt_key}'"
+
+    print("\n4. Testing draw_prompts_from_pool:")
+    drawn_prompts = generator.draw_prompts_from_pool(count=2)
+    print(f"   Drawn {len(drawn_prompts)} prompts from pool")
+    print(f"   Pool size after drawing: {len(generator.pool_prompts)}")
+
+    # Check drawn prompts have keys
+    for i, prompt_dict in enumerate(drawn_prompts):
+        prompt_key = list(prompt_dict.keys())[0]
+        prompt_text = prompt_dict[prompt_key]
+        print(f"   Drawn prompt {i+1}: Key='{prompt_key}', Text='{prompt_text[:30]}...'")
+
+    print("\n" + "="*60)
+    print("✅ All tests passed! New format works correctly.")
+    print("\nSummary:")
+    print("- AI returns prompts as a JSON list (no keys)")
+    print("- _parse_ai_response returns List[str]")
+    print("- add_prompts_to_pool generates keys locally (poolprompt000, poolprompt001, etc.)")
+    print("- draw_prompts_from_pool returns List[Dict[str, str]] with generated keys")
+
+    return True
+
+if __name__ == "__main__":
+    test_new_format()
+
+