Add tests for boolean iValues

Author: iceychris

build_dummy_model.py

@@ -29,10 +29,18 @@ def identity_tensor(self, x: torch.Tensor):
     def identity_long(self, x: int):
         return x
 
+    @torch.jit.export
+    def identity_bool(self, x: bool):
+        return x
+
     @torch.jit.export
     def identity_list(self, x: List[int]):
         return x
 
+    @torch.jit.export
+    def identity_bool_list(self, x: List[bool]):
+        return x
+
     @torch.jit.export
     def identity_tuple(self, x: Tuple[int]):
         return x

src/iosX64Test/kotlin/de/voize/pytorch_lite_multiplatform/TorchModuleIOSTest.kt

@@ -188,6 +188,52 @@ class TorchModuleIOSTest {
         }
     }
 
+    @Test
+    fun testIdentityBool() {
+        plmScoped {
+            val module = TorchModule(localModulePath)
+            val input = IValue.from(false)
+            val output = module.runMethod(
+                "identity_bool",
+                input
+            )
+            assertEquals(false, output.toBool())
+
+            val input2 = IValue.from(true)
+            val output2 = module.runMethod(
+                "identity_bool",
+                input2
+            )
+            assertEquals(true, output2.toBool())
+        }
+    }
+
+    @Test
+    fun testIdentityBoolList() {
+        plmScoped {
+            val module = TorchModule(localModulePath)
+            val input = IValue.listFrom(IValue.from(true), IValue.from(false))
+            val output = module.runMethod(
+                "identity_bool_list",
+                input
+            )
+            assertEquals(listOf(true, false), output.toBoolList())
+        }
+    }
+
+    @Test
+    fun testIdentityBoolList2() {
+        plmScoped {
+            val module = TorchModule(localModulePath)
+            val input = IValue.listFrom(true, false, scope = this)
+            val output = module.runMethod(
+                "identity_bool_list",
+                input
+            )
+            assertEquals(listOf(true, false), output.toBoolList())
+        }
+    }
+
     @Test
     fun testIdentityTensor() {
         plmScoped {