Convert example 04-enum to use JUnit

1 files changed, 22 insertions, 0 deletions

diff --git a/examples/04-enum/output/java-junit/src/test/java/org/zwobble/example/NumbersTests.java b/examples/04-enum/output/java-junit/src/test/java/org/zwobble/example/NumbersTests.java
new file mode 100644
index 0000000..95b85a4
--- /dev/null
+++ b/examples/04-enum/output/java-junit/src/test/java/org/zwobble/example/NumbersTests.java
@@ -0,0 +1,22 @@
+package org.zwobble.example;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import org.junit.jupiter.api.Test;
+import org.zwobble.example.types.numbers.SmallNumber;
+
+public class NumbersTests {
+    @Test
+    public void canEvalSmallNumbers() {
+        assertEquals(eval(SmallNumber.ZERO), 0);
+        assertEquals(eval(SmallNumber.ONE), 1);
+        assertEquals(eval(SmallNumber.TWO), 2);
+    }
+
+    private int eval(SmallNumber smallNumber) {
+        return switch (smallNumber) {
+            case ZERO -> 0;
+            case ONE -> 1;
+            case TWO -> 2;
+        };
+    }
+}