3 files changed, 52 insertions, 3 deletions
diff --git a/tests/001.sh b/tests/001.sh
index 9245cf6..70e00f5 100644
--- a/tests/001.sh
+++ b/tests/001.sh
@@ -5,10 +5,17 @@ int main() {
 EOF
 
 "$p4dcc" < main.c > main.s
+if [[ $? -ne 0 ]]; then
+    cat main.s >&2
+    exit 1
+fi
 gcc -o a.out main.s
-set +e
 ./a.out
 exit_code=$?
-set -e
 
-test $exit_code -eq 42
+expected=42
+
+if [[ ! $exit_code -eq $expected ]]; then
+    echo "expected $expected, but $exit_code" >&2
+    exit 1
+fi
diff --git a/tests/002.sh b/tests/002.sh
new file mode 100644
index 0000000..65d3d69
--- /dev/null
+++ b/tests/002.sh
@@ -0,0 +1,21 @@
+cat <<'EOF' > main.c
+int main() {
+    return 5+20-4;
+}
+EOF
+
+"$p4dcc" < main.c > main.s
+if [[ $? -ne 0 ]]; then
+    cat main.s >&2
+    exit 1
+fi
+gcc -o a.out main.s
+./a.out
+exit_code=$?
+
+expected=21
+
+if [[ ! $exit_code -eq $expected ]]; then
+    echo "expected $expected, but $exit_code" >&2
+    exit 1
+fi
diff --git a/tests/003.sh b/tests/003.sh
new file mode 100644
index 0000000..e99b7e1
--- /dev/null
+++ b/tests/003.sh
@@ -0,0 +1,21 @@
+cat <<'EOF' > main.c
+int main() {
+    return 2*3+4*5;
+}
+EOF
+
+"$p4dcc" < main.c > main.s
+if [[ $? -ne 0 ]]; then
+    cat main.s >&2
+    exit 1
+fi
+gcc -o a.out main.s
+./a.out
+exit_code=$?
+
+expected=26
+
+if [[ ! $exit_code -eq $expected ]]; then
+    echo "expected $expected, but $exit_code" >&2
+    exit 1
+fi