1 files changed, 38 insertions, 0 deletions
diff --git a/tests/programs/o3push.zc b/tests/programs/o3push.zc
new file mode 100644
index 0000000..e5f0090
--- /dev/null
+++ b/tests/programs/o3push.zc
@@ -0,0 +1,38 @@
+// push in o3 code: the quick path when there is room, dpush when there is not
+o3 int fill([]int a, int n) {
+	for (int i = 0; i < n; i++) push(a, i * 3);
+	return len(a);
+}
+o3 int sum([]int a) {
+	int s = 0;
+	for (int i = 0; i < len(a); i++) s += a[i];
+	return s;
+}
+o3 float fsum([]float a) {
+	float s = 0.0;
+	for (int i = 0; i < len(a); i++) s += a[i];
+	return s;
+}
+o3 int ffill([]float a, int n) {
+	for (int i = 0; i < n; i++) push(a, 0.5 * i);
+	return len(a);
+}
+int main() {
+	[]int a = new([]int, 0);
+	put_n(fill(a, 1000)); put_c(' ');
+	put_n(sum(a)); put_c(' ');
+	put_n(a[999]); put_c(' ');
+	put_n(fill(a, 5)); put_c(' ');
+	put_n(cap(a) >= 1005); put_c(' ');
+	[]float f = new([]float, 0);
+	ffill(f, 10);
+	put_n(len(f)); put_c(' '); put_n(fsum(f) == 22.5); put_c(' ');
+	put_n(drain(a)); put_c(' '); put_n(len(a)); put_c(10);
+	free(d*a); free(d*f);
+	return 0;
+}
+o3 int drain([]int a) {
+	int s = 0;
+	while (len(a) > 0) s += pop(a);
+	return s;
+}