test with enum with just one variant

Author: xunilrj

test/src/e2e_vm_tests/test_programs/should_pass/language/const_generics/src/main.sw

@@ -33,12 +33,24 @@ impl<T, const N: u64> S<T, N> {
     }
 }
 
-enum E<T, const N: u64> {
+// Enum with just one variant
+enum OneVariant<const N: u64> {
+    A: [u64; N],
+}
+
+impl<const N: u64> OneVariant<N> {
+    pub fn return_n(self) -> u64 {
+        N
+    }
+}
+
+// Enum with more than one variant
+enum TwoVariants<T, const N: u64> {
     Nothing: (),
     Array: [T; N]
 }
 
-impl<T, const N: u64> E<T, N> {
+impl<T, const N: u64> TwoVariants<T, N> {
     pub fn len_xxx2(self) -> u64 {
         N
     }
@@ -56,7 +68,15 @@ fn main(a: [u64; 2]) {
     let s: S<u64, 3> = S { };
     let _ = __dbg(s.len_xxx());
 
-    let e: E<u64, 3> = E::<u64, 3>::Nothing;
+    // Check enum with just one variant, with
+    // all types explicit
+    let e: OneVariant<3> = OneVariant::<3>::A([1u64, 2u64, 3u64]);
+    assert(e.return_n() == 3);
+    let _ = __dbg(e);
+
+    // Check enum with more than one variant, with
+    // all types explicit
+    let e: TwoVariants<u64, 3> = TwoVariants::<u64, 3>::Nothing;
     let _ = __dbg(e);
     let b = e.len_xxx2();
     assert(b == 3);

test/src/e2e_vm_tests/test_programs/should_pass/language/const_generics/stdout.snap

@@ -7,15 +7,16 @@ output:
     Building test/src/e2e_vm_tests/test_programs/should_pass/language/const_generics
    Compiling library std (sway-lib-std)
    Compiling script const_generics (test/src/e2e_vm_tests/test_programs/should_pass/language/const_generics)
-    Finished debug [unoptimized + fuel] target(s) [3.664 KB] in ???
+    Finished debug [unoptimized + fuel] target(s) [4.472 KB] in ???
      Running 1 test, filtered 0 tests
 
 tested -- const_generics
 
-      test main_test ... ok (???, 3212 gas)
-[src/main.sw:48:13] a = [0, 1]
-[src/main.sw:57:13] s.len_xxx() = 3
-[src/main.sw:60:13] e = Nothing
+      test main_test ... ok (???, 6377 gas)
+[src/main.sw:60:13] a = [0, 1]
+[src/main.sw:69:13] s.len_xxx() = 3
+[src/main.sw:75:13] e = OneVariant([1, 2, 3])
+[src/main.sw:80:13] e = Nothing
 
 test result: OK. 1 passed; 0 failed; finished in ???