add dimensionality check for index operators

Rue Yokaze · Rue Yokaze · commit 46b120ca8132 · 2017-06-21T23:22:08.000+09:00
diff --git a/python_multi_array.cpp b/python_multi_array.cpp
@@ -203,6 +203,10 @@ namespace python_multi_array
             }
         }
         //  assume idx to be a list or a tuple
+        if (N != python::len(idx))
+        {
+            throw std::invalid_argument("idx");
+        }
         size_t s[N];
         for (size_t i = 0; i < N; ++i)
         {
@@ -230,6 +234,10 @@ namespace python_multi_array
             }
         }
         //  assume idx to be a list or a tuple
+        if (N != python::len(idx))
+        {
+            throw std::invalid_argument("idx");
+        }
         size_t s[N];
         for (size_t i = 0; i < N; ++i)
         {

Original file line number	Diff line number	Diff line change
`@@ -203,6 +203,10 @@ namespace python_multi_array`
`203`	`203`	`}`
`204`	`204`	`}`
`205`	`205`	`// assume idx to be a list or a tuple`
	`206`	`+ if (N != python::len(idx))`
	`207`	`+ {`
	`208`	`+ throw std::invalid_argument("idx");`
	`209`	`+ }`
`206`	`210`	`size_t s[N];`
`207`	`211`	`for (size_t i = 0; i < N; ++i)`
`208`	`212`	`{`
`@@ -230,6 +234,10 @@ namespace python_multi_array`
`230`	`234`	`}`
`231`	`235`	`}`
`232`	`236`	`// assume idx to be a list or a tuple`
	`237`	`+ if (N != python::len(idx))`
	`238`	`+ {`
	`239`	`+ throw std::invalid_argument("idx");`
	`240`	`+ }`
`233`	`241`	`size_t s[N];`
`234`	`242`	`for (size_t i = 0; i < N; ++i)`
`235`	`243`	`{`