From e3d69a0cf53e836736b8660a2a9f040aa25f2c91 Mon Sep 17 00:00:00 2001 From: Nathan VanBenschoten Date: Wed, 2 Sep 2026 17:28:35 -0400 Subject: [PATCH] tests: accept 1.28GB query bytes floor CI against current servers still returns 256MB, while servers with the new pricing floor return 1.28GB. Co-authored-by: Cursor --- tests/custom/test_vectors.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/custom/test_vectors.py b/tests/custom/test_vectors.py index 5e3ed80c..75407a3e 100644 --- a/tests/custom/test_vectors.py +++ b/tests/custom/test_vectors.py @@ -10,7 +10,6 @@ from turbopuffer.types import ( Row, Vector, - QueryBilling, VectorEncoding, NamespaceWriteParams, NamespaceQueryResponse, @@ -203,10 +202,9 @@ def check_results(vector_set: NamespaceQueryResponse, expected: List[Row]): include_attributes=["hello", "vector"], ) check_results(vector_set, expected) - assert vector_set.billing == QueryBilling( - billable_logical_bytes_queried=256000000, - billable_logical_bytes_returned=105, - ) + assert vector_set.billing.billable_logical_bytes_returned == 105 + # 256MB was the old query floor; 1.28GB is current pricing. + assert vector_set.billing.billable_logical_bytes_queried in (256000000, 1280000000) # Test query with dict vector_set = ns.query(