Skip to content

Commit 9afa3fc

Browse files
Rusty Conoverclaude
andcommitted
test: satisfy mypy in catalog_contents and ts_types tests
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
1 parent 7e8623f commit 9afa3fc

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

‎tests/catalog/test_catalog_contents.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -520,14 +520,14 @@ def test_item_encoding_sorts_map_keys() -> None:
520520
from vgi.protocol import _item_ipc_bytes
521521

522522
a = SchemaInfo(
523-
attach_opaque_data=b"x",
523+
attach_opaque_data=AttachOpaqueData(b"x"),
524524
path=["s"],
525525
comment=None,
526526
tags={"b": "2", "a": "1"},
527527
estimated_object_count={"view": 0, "table": 3},
528528
)
529529
b = SchemaInfo(
530-
attach_opaque_data=b"x",
530+
attach_opaque_data=AttachOpaqueData(b"x"),
531531
path=["s"],
532532
comment=None,
533533
tags={"a": "1", "b": "2"},

‎tests/test_generated_ts_types.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import pytest
1919

2020
from vgi.codegen import ts_types
21+
from vgi.codegen._common import GeneratorError
2122
from vgi.codegen.ts_types import TS_RECORD_TYPES, build_model, emit
2223

2324

@@ -90,7 +91,7 @@ def test_unsupported_type_raises() -> None:
9091
"""An Arrow type with no mapping fails loudly instead of emitting a wrong type."""
9192
import pyarrow as pa
9293

93-
with pytest.raises(ts_types.GeneratorError):
94+
with pytest.raises(GeneratorError):
9495
ts_types._ts_type(pa.struct([pa.field("a", pa.int32())]), "X.a")
9596

9697

0 commit comments

Comments
 (0)