CallMeDaniel Claude Opus 4.6 (1M context) commited on
Commit
beda709
·
1 Parent(s): d1b7ccd

refactor: update test fixtures to use Pydantic models

Browse files

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Files changed (1) hide show
  1. tests/conftest.py +11 -9
tests/conftest.py CHANGED
@@ -3,6 +3,8 @@
3
  import pytest
4
  from pathlib import Path
5
 
 
 
6
 
7
  @pytest.fixture
8
  def tmp_output_dir(tmp_path):
@@ -25,20 +27,20 @@ def sample_history():
25
 
26
  @pytest.fixture
27
  def empty_design_state():
28
- """Empty design state dict."""
29
- return {}
30
 
31
 
32
  @pytest.fixture
33
  def populated_design_state():
34
  """Design state with some decisions already made."""
35
- return {
36
- "part_name": "servo_bracket",
37
- "material": "aluminum 6061",
38
- "dimensions": {"width": 60.0},
39
- "features": ["4x M4 holes"],
40
- "decisions": ["L-bracket form factor"],
41
- }
42
 
43
 
44
  class FakeLLMBackend:
 
3
  import pytest
4
  from pathlib import Path
5
 
6
+ from agents.design_state import DesignState
7
+
8
 
9
  @pytest.fixture
10
  def tmp_output_dir(tmp_path):
 
27
 
28
  @pytest.fixture
29
  def empty_design_state():
30
+ """Empty design state."""
31
+ return DesignState()
32
 
33
 
34
  @pytest.fixture
35
  def populated_design_state():
36
  """Design state with some decisions already made."""
37
+ return DesignState(
38
+ part_name="servo_bracket",
39
+ material="aluminum 6061",
40
+ dimensions={"width": 60.0},
41
+ features=["4x M4 holes"],
42
+ decisions=["L-bracket form factor"],
43
+ )
44
 
45
 
46
  class FakeLLMBackend: