Implement From for Rust variants

1 files changed, 5 insertions, 3 deletions

diff --git a/examples/02-sum/output/rust/src/shapes.rs b/examples/02-sum/output/rust/src/shapes.rs
index 31c3db9..d5b7334 100644
--- a/examples/02-sum/output/rust/src/shapes.rs
+++ b/examples/02-sum/output/rust/src/shapes.rs
@@ -6,12 +6,14 @@ pub enum Shape {
     Triangle(crate::shapes::Triangle),
 }
 
-impl Shape {
-    pub fn rectangle(value: crate::shapes::Rectangle) -> Self {
+impl ::std::convert::From<crate::shapes::Rectangle> for crate::shapes::Shape {
+    fn from(value: crate::shapes::Rectangle) -> Self {
         Self::Rectangle(value)
     }
+}
 
-    pub fn triangle(value: crate::shapes::Triangle) -> Self {
+impl ::std::convert::From<crate::shapes::Triangle> for crate::shapes::Shape {
+    fn from(value: crate::shapes::Triangle) -> Self {
         Self::Triangle(value)
     }
 }