ut

xadupre · xadupre · commit 49ef7ceea104 · 2023-11-23T01:13:20.000+01:00
diff --git a/_unittests/ut_light_api/test_light_api.py b/_unittests/ut_light_api/test_light_api.py
@@ -472,6 +472,23 @@ def test_if(self):
         got = ref.run(None, {"X": -x})
         self.assertEqualArray(np.array([0], dtype=np.int64), got[0])
 
+    def test_domain(self):
+        onx = (
+            start()
+            .vin("X")
+            .reshape((-1, 1))
+            .dom["ai.onnx.ml"].Normalizer(norm="L1")
+            .rename("Y")
+            .vout()
+            .to_onnx()
+        )
+        self.assertIsInstance(onx, ModelProto)
+        self.assertIn("Transpose", str(onx))
+        ref = ReferenceEvaluator(onx)
+        a = np.arange(10).astype(np.float32)
+        got = ref.run(None, {"X": a})[0]
+        self.assertEqualArray(a.reshape((-1, 1)).T, got)
+
 
 if __name__ == "__main__":
     TestLightApi().test_if()
