File size: 4,088 Bytes
e5f64b3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 | # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
#!/usr/bin/env python3
"""
Simple test showing how users will use CodingEnv.from_docker_image().
This is the simplest possible usage
"""
import sys
from pathlib import Path
# Add src to path
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
from coding_env import CodeAction, CodingEnv
def main():
"""Test CodingEnv.from_docker_image()."""
print("=" * 60)
print("CodingEnv.from_docker_image() Test")
print("=" * 60)
print()
try:
# This is what users will do - just one line!
print("Creating client from Docker image...")
print(" CodingEnv.from_docker_image('coding-env:latest')")
print()
client = CodingEnv.from_docker_image("coding-env:latest")
print("✓ Client created and container started!\n")
# Now use it like any other client
print("Testing the environment:")
print("-" * 60)
# Reset
print("\n1. Reset:")
result = client.reset()
print(f" stdout: {result.observation.stdout}")
print(f" stderr: {result.observation.stderr}")
print(f" exit_code: {result.observation.exit_code}")
# Get initial state
state = client.state()
print(f" State: episode_id={state.episode_id}, step_count={state.step_count}")
# Execute some Python code
print("\n2. Execute Python code:")
code_samples = [
"print('Hello, World!')",
"x = 5 + 3\nprint(f'Result: {x}')",
"import math\nprint(f'Pi is approximately {math.pi:.4f}')",
"# Multi-line calculation\nfor i in range(1, 4):\n print(f'{i} squared is {i**2}')",
]
for i, code in enumerate(code_samples, 1):
result = client.step(CodeAction(code=code))
print(f" {i}. Code: {code.replace(chr(10), '\\n')[:50]}...")
print(f" → stdout: {result.observation.stdout.strip()}")
print(f" → exit_code: {result.observation.exit_code}")
if result.observation.stderr:
print(f" → stderr: {result.observation.stderr}")
# Test error scenarios
print("\n3. Test error scenarios:")
error_samples = [
("Division by zero", "x = 1 / 0\nprint('Should not reach here')"),
("Undefined variable", "print(undefined_variable)"),
("Syntax error", "print('Hello'"),
]
for i, (description, code) in enumerate(error_samples, 1):
result = client.step(CodeAction(code=code))
print(f" {i}. {description}")
print(f" Code: {code.replace(chr(10), '\\n')[:40]}...")
print(f" → exit_code: {result.observation.exit_code}")
if result.observation.stderr:
# Truncate long error messages
error_msg = result.observation.stderr[:100]
if len(result.observation.stderr) > 100:
error_msg += "..."
print(f" → stderr: {error_msg}")
# Check final state
print("\n4. Check final state:")
state = client.state()
print(f" episode_id: {state.episode_id}")
print(f" step_count: {state.step_count}")
print(f" last_exit_code: {state.last_exit_code}")
print("\n" + "-" * 60)
print("\n✓ All operations successful!")
print()
print("Cleaning up...")
client.close()
print("✓ Container stopped and removed")
print()
print("=" * 60)
print("Test completed successfully! 🎉")
print("=" * 60)
return True
except Exception as e:
print(f"\n❌ Test failed: {e}")
import traceback
traceback.print_exc()
return False
if __name__ == "__main__":
success = main()
exit(0 if success else 1)
|