102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
import unittest
|
|
|
|
from sglang.srt.utils.field_validators import (
|
|
validate_list_i64_1d,
|
|
validate_optional_list_i64_1d_2d,
|
|
)
|
|
from sglang.test.ci.ci_register import register_cpu_ci
|
|
from sglang.test.test_utils import CustomTestCase
|
|
|
|
register_cpu_ci(est_time=2, suite="base-a-test-cpu")
|
|
|
|
|
|
class TestValidateListI64_1d(CustomTestCase):
|
|
"""`validate_list_i64_1d` rejects anything that array('q', v) can't accept."""
|
|
|
|
def test_accept_int_list(self):
|
|
v = [1, 2, 3]
|
|
self.assertIs(validate_list_i64_1d(v), v)
|
|
|
|
def test_accept_empty_list(self):
|
|
v = []
|
|
self.assertIs(validate_list_i64_1d(v), v)
|
|
|
|
def test_accept_int64_boundaries(self):
|
|
for v in ([-(2**63)], [2**63 - 1], [0]):
|
|
self.assertIs(validate_list_i64_1d(v), v)
|
|
|
|
def test_reject_none(self):
|
|
with self.assertRaisesRegex(ValueError, "must not be None"):
|
|
validate_list_i64_1d(None)
|
|
|
|
def test_reject_non_list(self):
|
|
for v in ((1, 2, 3), "abc", {1: 2}, 42, 3.14):
|
|
with self.subTest(v=v):
|
|
with self.assertRaisesRegex(ValueError, "must be list"):
|
|
validate_list_i64_1d(v)
|
|
|
|
def test_reject_non_int_first_element(self):
|
|
for v in ([1.5, 2, 3], ["a", "b"], [None, 1]):
|
|
with self.subTest(v=v):
|
|
with self.assertRaisesRegex(ValueError, "elements must be int"):
|
|
validate_list_i64_1d(v)
|
|
|
|
def test_reject_overflow_int64(self):
|
|
# 2**63 overflows signed int64.
|
|
with self.assertRaisesRegex(ValueError, "non-int64 element"):
|
|
validate_list_i64_1d([0, 2**63])
|
|
|
|
def test_reject_non_int_later_element(self):
|
|
# First-element fast path passes but C loop rejects later float.
|
|
with self.assertRaisesRegex(ValueError, "non-int64 element"):
|
|
validate_list_i64_1d([1, 2, 3.5])
|
|
|
|
|
|
class TestValidateOptionalListI64_1d_2d(CustomTestCase):
|
|
"""`validate_optional_list_i64_1d_2d` accepts None | [] | list[int] | list[list[int]]."""
|
|
|
|
def test_accept_none(self):
|
|
self.assertIsNone(validate_optional_list_i64_1d_2d(None))
|
|
|
|
def test_accept_empty_list(self):
|
|
v = []
|
|
self.assertIs(validate_optional_list_i64_1d_2d(v), v)
|
|
|
|
def test_accept_1d_int_list(self):
|
|
v = [1, 2, 3]
|
|
self.assertIs(validate_optional_list_i64_1d_2d(v), v)
|
|
|
|
def test_accept_2d_int_list(self):
|
|
v = [[1, 2], [3, 4, 5]]
|
|
self.assertIs(validate_optional_list_i64_1d_2d(v), v)
|
|
|
|
def test_accept_2d_with_empty_row(self):
|
|
v = [[], [1, 2]]
|
|
self.assertIs(validate_optional_list_i64_1d_2d(v), v)
|
|
|
|
def test_reject_non_list_top_level(self):
|
|
for v in ((1, 2), "abc", 42, 3.14, {1: 2}):
|
|
with self.subTest(v=v):
|
|
with self.assertRaisesRegex(ValueError, "must be list or null"):
|
|
validate_optional_list_i64_1d_2d(v)
|
|
|
|
def test_reject_mixed_first_element_type(self):
|
|
with self.assertRaisesRegex(ValueError, "elements must be int or list"):
|
|
validate_optional_list_i64_1d_2d([1.5, 2.5])
|
|
|
|
def test_reject_overflow_in_1d(self):
|
|
with self.assertRaisesRegex(ValueError, "non-int64 element"):
|
|
validate_optional_list_i64_1d_2d([0, 2**63])
|
|
|
|
def test_reject_bad_row_in_2d_reports_index(self):
|
|
with self.assertRaisesRegex(ValueError, "row 1:"):
|
|
validate_optional_list_i64_1d_2d([[1, 2], [3, "x"]])
|
|
|
|
def test_reject_overflow_in_2d_reports_index(self):
|
|
with self.assertRaisesRegex(ValueError, "row 0:.*non-int64"):
|
|
validate_optional_list_i64_1d_2d([[2**63], [1, 2]])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|