test

Author: andrewtruong

tests/trace/test_anonymous_ops.py

@@ -2,28 +2,6 @@
 from weave.trace_server.trace_server_interface import CallsQueryReq
 
 
-def test_named_op(client: weave_client.WeaveClient) -> str:
-    call = client.create_call("anonymous_op", {"a": 1})
-    client.finish_call(call, {"c": 3}, None)
-
-    call_res = client.server.calls_query(
-        CallsQueryReq(
-            project_id=client._project_id(),
-        )
-    )
-    calls = call_res.calls
-
-    assert len(calls) == 1
-    call = calls[0]
-    assert (
-        call.op_name
-        == "weave:///shawn/test-project/op/anonymous_op:IMZjfLeSAoLTYu8QLPrVVXrNIFX6jZlxAYAeRDEEezE"
-    )
-    assert call.inputs == {"a": 1}
-    assert call.output == {"c": 3}
-    assert call.exception is None
-
-
 def test_anonymous_op(client: weave_client.WeaveClient) -> str:
     call = client.create_call("anonymous_op", {"a": 1})
     client.finish_call(call, {"c": 3}, None)

tests/trace/test_op_argument_forms.py

@@ -450,23 +450,6 @@ def my_op() -> int:
     assert res.calls[0].inputs == {}
 
 
-def test_args_empty(client):
-    @weave.op
-    def my_op() -> int:
-        return 1
-
-    my_op()
-
-    res = client.server.calls_query(
-        tsi.CallsQueryReq(
-            project_id=client._project_id(),
-        )
-    )
-
-    assert res.calls[0].op_name == my_op.ref.uri()
-    assert res.calls[0].inputs == {}
-
-
 def test_args_concrete(client):
     @weave.op
     def my_op(val):

tests/trace/test_op_generators.py

@@ -203,44 +203,6 @@ async def test_async_generator_with_custom_accumulator(client):
     assert calls[0].output == [0, 1, 2]
 
 
-@weave.op(accumulator=list_accumulator)
-def gen_with_decorator_accumulator(x: int) -> Generator[int, None, None]:
-    yield from range(x)
-
-
-def test_generator_with_decorator_accumulator(client):
-    # Call the generator with the accumulator from the decorator
-    res = gen_with_decorator_accumulator(3)
-
-    # The generator still works as expected
-    assert list(res) == [0, 1, 2]
-
-    # Get the call and check its output
-    calls = client.get_calls()
-    assert len(calls) == 1
-    assert calls[0].output == [0, 1, 2]
-
-
-@weave.op(accumulator=async_list_accumulator)
-async def async_gen_with_decorator_accumulator(x: int) -> AsyncGenerator[int, None]:
-    for i in range(x):
-        yield i
-
-
-@pytest.mark.asyncio
-async def test_async_generator_with_decorator_accumulator(client):
-    # Call the generator with the accumulator from the decorator
-    res = async_gen_with_decorator_accumulator(3)
-
-    # The generator still works as expected
-    assert [item async for item in res] == [0, 1, 2]
-
-    # Get the call and check its output
-    calls = client.get_calls()
-    assert len(calls) == 1
-    assert calls[0].output == [0, 1, 2]
-
-
 def test_nested_generator_multiple_iterations(client):
     """Test that nested generators work correctly when called multiple times.
 

tests/trace_server_bindings/test_http_behavior_remote.py

@@ -15,7 +15,6 @@
 import httpx
 import pytest
 import tenacity
-from pydantic import ValidationError
 
 from tests.trace_server_bindings.conftest import (
     generate_end,
@@ -88,12 +87,6 @@ def test_400_no_retry(mock_post, unbatched_server):
         unbatched_server.call_start(tsi.CallStartReq(start=start))
 
 
-def test_invalid_no_retry(unbatched_server):
-    """Test that validation errors are not retried."""
-    with pytest.raises(ValidationError):
-        unbatched_server.call_start(tsi.CallStartReq(start={"invalid": "broken"}))
-
-
 @patch("weave.utils.http_requests.post")
 def test_calls_complete_batch_endpoint_and_payload(mock_post, monkeypatch):
     """Send calls_complete batches to the v2 endpoint with correct payload."""