Merge pull request xtensor-stack#180 from PerretB/test_call_with_incorrect_type

JohanMabille · web-flow · commit fef5b93d9868 · 2018-11-16T13:27:19.000+01:00
Add test for function call with non array arguments
diff --git a/test_python/main.cpp b/test_python/main.cpp
@@ -239,4 +239,7 @@ PYBIND11_MODULE(xtensor_python_test, m)
             [](C & self) -> C::array_type & { return self.array(); }
         )
     ;
+
+    m.def("simple_array", [](xt::pyarray<int>) { return 1; } );
+    m.def("simple_tensor", [](xt::pytensor<int, 1>) { return 2; } );
 }
diff --git a/test_python/test_pyarray.py b/test_python/test_pyarray.py
@@ -147,6 +147,13 @@ def test_col_row_major(self):
         xt.col_major_array(varF)
         xt.col_major_array(varF[:, :, 0]) # still col major!
 
+    def test_bad_argument_call(self):
+        with self.assertRaises(TypeError):
+            xt.simple_array("foo")
+
+        with self.assertRaises(TypeError):
+            xt.simple_tensor("foo")
+
 class AttributeTest(TestCase):
 
     def setUp(self):

Original file line number	Diff line number	Diff line change
`@@ -239,4 +239,7 @@ PYBIND11_MODULE(xtensor_python_test, m)`
`239`	`239`	`[](C & self) -> C::array_type & { return self.array(); }`
`240`	`240`	`)`
`241`	`241`	`;`
	`242`	`+`
	`243`	`+ m.def("simple_array", [](xt::pyarray<int>) { return 1; } );`
	`244`	`+ m.def("simple_tensor", [](xt::pytensor<int, 1>) { return 2; } );`
`242`	`245`	`}`