Convert Rust examples to use cargo test

2 files changed, 29 insertions, 16 deletions

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 })
-}