Add tests

python · JukkaL · Apr 1, 2026 · Apr 1, 2026 · Apr 1, 2026 · Apr 1, 2026
commit 739ee98730c4ad73bd09fa731e0645fbc9e8cb71
diff --git a/mypyc/test-data/irbuild-vec-i64.test b/mypyc/test-data/irbuild-vec-i64.test
@@ -864,3 +864,38 @@ L0:
     r4 = r3 + 8
     keep_alive r0
     return r0
+
+[case testVecI64CreateFromListMultiplyWithCap]
+from librt.vecs import vec
+from mypy_extensions import i64
+
+def repeated_with_cap(n: i64) -> vec[i64]:
+    return vec[i64]([3] * n, capacity=10)
+[out]
+def repeated_with_cap(n):
+    n :: i64
+    r0 :: vec[i64]
+    r1 :: object
+    r2 :: ptr
+    r3 :: i64
+    r4, r5 :: ptr
+    r6 :: bit
+    r7 :: ptr
+L0:
+    r0 = VecI64Api.alloc(n, 10)
+    r1 = r0.buf
+    r2 = get_element_ptr r1 items :: VecI64BufObject
+    r3 = n * 8
+    r4 = r2 + r3
+    r5 = r2
+L1:
+    r6 = r5 < r4 :: unsigned
+    if r6 goto L2 else goto L3 :: bool
+L2:
+    set_mem r5, 3 :: i64*
+    r7 = r5 + 8
+    r5 = r7
+    goto L1
+L3:
+    keep_alive r0
+    return r0
diff --git a/mypyc/test-data/run-vecs-t.test b/mypyc/test-data/run-vecs-t.test
@@ -460,6 +460,21 @@ def test_cap_with_initializer() -> None:
     assert v[0] == 'a'
     assert v[1] == 'b'
 
+def test_cap_from_comprehension_and_iterable() -> None:
+    v = vec[str]([str(i) for i in range(2)], capacity=3)
+    old = v
+    v = append(v, '2')
+    v[0] = 'x'
+    assert old[0] == 'x'
+    assert v == vec[str](['x', '1', '2'])
+
+    v = vec[str]((str(i) for i in range(2)), capacity=3)
+    old = v
+    v = append(v, '2')
+    v[0] = 'y'
+    assert old[0] == 'y'
+    assert v == vec[str](['y', '1', '2'])
+
 def test_cap_variable(n: i64 = 3) -> None:
     v = vec[str](capacity=n)
     assert len(v) == 0