4 files changed, 48 insertions, 28 deletions
diff --git a/examples/02-sum/output/java/src/main/java/org/zwobble/example/Shapes.java b/examples/02-sum/output/java/src/main/java/org/zwobble/example/Shapes.java
new file mode 100644
index 0000000..a6a468f
--- /dev/null
+++ b/examples/02-sum/output/java/src/main/java/org/zwobble/example/Shapes.java
@@ -0,0 +1,17 @@
+package org.zwobble.example;
+
+import org.zwobble.example.types.shapes.Shape;
+import org.zwobble.example.types.shapes.Rectangle;
+import org.zwobble.example.types.shapes.Triangle;
+
+public class Shapes {
+    public static int area(Shape shape) {
+        return switch (shape) {
+            case Rectangle(var width, var height) ->
+                width * height;
+
+            case Triangle(var width, var height) ->
+                width * height / 2;
+        };
+    }
+}
diff --git a/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java b/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
index 2554ca5..893a544 100644
--- a/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
+++ b/examples/02-sum/output/java/src/test/java/org/zwobble/example/ShapesTests.java
@@ -10,24 +10,14 @@ public class ShapesTests {
     @Test
     public void rectangle() {
         Shape rectangle = new Rectangle(10, 25);
-        int area = area(rectangle);
+        int area = Shapes.area(rectangle);
         assertEquals(area, 250);
     }
 
     @Test
     public void triangle() {
         Shape triangle = new Triangle(10, 25);
-        int area = area(triangle);
+        int area = Shapes.area(triangle);
         assertEquals(area, 125);
     }
-
-    private static int area(Shape shape) {
-        return switch (shape) {
-            case Rectangle(var width, var height) ->
-                width * height;
-
-            case Triangle(var width, var height) ->
-                width * height / 2;
-        };
-    }
 }
diff --git a/examples/02-sum/output/rust/src/lib.rs b/examples/02-sum/output/rust/src/lib.rs
new file mode 100644
index 0000000..f961002
--- /dev/null
+++ b/examples/02-sum/output/rust/src/lib.rs
@@ -0,0 +1,29 @@
+mod shapes;
+
+use shapes::{Rectangle, Shape, Triangle};
+
+fn area(shape: &Shape) -> i32 {
+    match shape {
+        Shape::Rectangle(rectangle) => rectangle.width * rectangle.height,
+        Shape::Triangle(triangle) => triangle.width * triangle.height / 2,
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use super::shapes::{Rectangle, Shape, Triangle};
+
+    #[test]
+    fn rectangle() {
+        let rectangle = Shape::Rectangle(Rectangle { width: 10, height: 25 });
+        let area = super::area(&rectangle);
+        assert_eq!(area, 250);
+    }
+
+    #[test]
+    fn triangle() {
+        let triangle = Shape::Triangle(Triangle { width: 10, height: 25 });
+        let area = super::area(&triangle);
+        assert_eq!(area, 125);
+    }
+}
diff --git a/examples/02-sum/output/rust/src/main.rs b/examples/02-sum/output/rust/src/main.rs
deleted file mode 100644
index 010ea2a..0000000
--- a/examples/02-sum/output/rust/src/main.rs
+++ /dev/null
@@ -1,16 +0,0 @@
-mod shapes;
-
-use shapes::{Rectangle, Shape, Triangle};
-
-fn main() {
-    let shape = create_shape();
-
-    let area = match shape {
-        Shape::Rectangle(rectangle) => rectangle.width * rectangle.height,
-        Shape::Triangle(triangle) => triangle.width * triangle.height / 2,
-    };
-}
-
-fn create_shape() -> Shape {
-    Shape::Rectangle(Rectangle { width: 10, height: 25 })
-}