Test more

JukkaL · JukkaL · commit e49c42b9dcea · 2026-04-23T16:44:33.000+01:00
diff --git a/mypyc/test-data/run-vecs-i64.test b/mypyc/test-data/run-vecs-i64.test
@@ -488,3 +488,27 @@ def test_tuple() -> None:
     assert b == t[1]
     with assertRaises(RuntimeError):
         ftuple(True)
+
+def test_cap_empty() -> None:
+    v = vec[i64](cap=3)
+    assert len(v) == 0
+    assert v == vec[i64]()
+
+def test_cap_with_initializer() -> None:
+    v = vec[i64]([1, 2], cap=3)
+    assert len(v) == 2
+    assert v[0] == 1
+    assert v[1] == 2
+
+def test_cap_variable(n: i64 = 3) -> None:
+    v = vec[i64](cap=n)
+    assert len(v) == 0
+    v = append(v, 5)
+    assert v[0] == 5
+
+def test_cap_buffer_reuse() -> None:
+    v = vec[i64]([1, 2], cap=3)
+    old = v
+    v = append(v, 3)
+    v[0] = 99
+    assert old[0] == 99  # shared buffer
diff --git a/mypyc/test-data/run-vecs-misc.test b/mypyc/test-data/run-vecs-misc.test
@@ -236,3 +236,30 @@ def test_nested_misc() -> None:
     v[0][0] = 3.5
     assert v[0] == vec[float]([3.5])
     assert v[:5] == vec[vec[float]]([vec[float]([3.5])])
+
+def test_cap_misc() -> None:
+    v_f = vec[float](cap=3)
+    assert len(v_f) == 0
+    v_f = append(v_f, 1.5)
+    assert v_f[0] == 1.5
+
+    v_u8 = vec[u8]([1, 2], cap=5)
+    assert len(v_u8) == 2
+    assert v_u8[0] == 1
+
+    v_i16 = vec[i16](cap=2)
+    v_i16 = append(v_i16, i16(10))
+    assert v_i16[0] == 10
+
+    v_i32 = vec[i32]([i32(7)], cap=3)
+    assert v_i32[0] == 7
+
+    v_bool = vec[bool](cap=4)
+    assert len(v_bool) == 0
+    v_bool = append(v_bool, True)
+    assert v_bool[0] == True
+
+    v_nested = vec[vec[float]](cap=2)
+    assert len(v_nested) == 0
+    v_nested = append(v_nested, vec[float]([1.0]))
+    assert v_nested[0] == vec[float]([1.0])
diff --git a/mypyc/test-data/run-vecs-nested.test b/mypyc/test-data/run-vecs-nested.test
@@ -518,3 +518,27 @@ def test_append_with_alias() -> None:
     assert len(alias) == 1
     assert v[0] == vec[i64]([1, 2, 3])
     assert len(v) == 4
+
+def test_cap_empty() -> None:
+    v = vec[vec[i64]](cap=3)
+    assert len(v) == 0
+    assert v == vec[vec[i64]]()
+
+def test_cap_with_initializer() -> None:
+    v = vec[vec[i64]]([vec[i64]([1]), vec[i64]([2])], cap=3)
+    assert len(v) == 2
+    assert v[0] == vec[i64]([1])
+    assert v[1] == vec[i64]([2])
+
+def test_cap_variable(n: i64 = 3) -> None:
+    v = vec[vec[str]](cap=n)
+    assert len(v) == 0
+    v = append(v, vec[str](["x"]))
+    assert v[0] == vec[str](["x"])
+
+def test_cap_buffer_reuse() -> None:
+    v = vec[vec[i64]]([vec[i64]([1]), vec[i64]([2])], cap=3)
+    old = v
+    v = append(v, vec[i64]([3]))
+    v[0] = vec[i64]([99])
+    assert old[0] == vec[i64]([99])  # shared buffer
diff --git a/mypyc/test-data/run-vecs-t.test b/mypyc/test-data/run-vecs-t.test
@@ -448,3 +448,27 @@ def test_append_with_alias() -> None:
     assert len(alias) == 1
     assert v[0] == 'a'
     assert len(v) == 4
+
+def test_cap_empty() -> None:
+    v = vec[str](cap=3)
+    assert len(v) == 0
+    assert v == vec[str]()
+
+def test_cap_with_initializer() -> None:
+    v = vec[str](['a', 'b'], cap=3)
+    assert len(v) == 2
+    assert v[0] == 'a'
+    assert v[1] == 'b'
+
+def test_cap_variable(n: i64 = 3) -> None:
+    v = vec[str](cap=n)
+    assert len(v) == 0
+    v = append(v, 'x')
+    assert v[0] == 'x'
+
+def test_cap_buffer_reuse() -> None:
+    v = vec[str](['a', 'b'], cap=3)
+    old = v
+    v = append(v, 'c')
+    v[0] = 'z'
+    assert old[0] == 'z'  # shared buffer
