FIX: Add debug assertions for the axis in RemoveAxis

bluss · bluss · commit 3ce3059122bc · 2018-11-18T08:52:27.000+01:00
See previous commit, since the trait is not intended to be user facing,
use debug assertions.
diff --git a/src/dimension/remove_axis.rs b/src/dimension/remove_axis.rs
@@ -19,7 +19,10 @@ pub trait RemoveAxis : Dimension {
 
 impl RemoveAxis for Dim<[Ix; 1]> {
     #[inline]
-    fn remove_axis(&self, _: Axis) -> Ix0 { Ix0() }
+    fn remove_axis(&self, axis: Axis) -> Ix0 {
+        debug_assert!(axis.index() < self.ndim());
+        Ix0()
+    }
 }
 
 impl RemoveAxis for Dim<[Ix; 2]> {
@@ -38,6 +41,7 @@ macro_rules! impl_remove_axis_array(
         {
             #[inline]
             fn remove_axis(&self, axis: Axis) -> Self::Smaller {
+                debug_assert!(axis.index() < self.ndim());
                 let mut tup = Dim([0; $n - 1]);
                 {
                     let mut it = tup.slice_mut().iter_mut();

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,10 @@ pub trait RemoveAxis : Dimension {`
`19`	`19`
`20`	`20`	`impl RemoveAxis for Dim<[Ix; 1]> {`
`21`	`21`	`#[inline]`
`22`		`- fn remove_axis(&self, _: Axis) -> Ix0 { Ix0() }`
	`22`	`+ fn remove_axis(&self, axis: Axis) -> Ix0 {`
	`23`	`+ debug_assert!(axis.index() < self.ndim());`
	`24`	`+ Ix0()`
	`25`	`+ }`
`23`	`26`	`}`
`24`	`27`
`25`	`28`	`impl RemoveAxis for Dim<[Ix; 2]> {`
`@@ -38,6 +41,7 @@ macro_rules! impl_remove_axis_array(`
`38`	`41`	`{`
`39`	`42`	`#[inline]`
`40`	`43`	`fn remove_axis(&self, axis: Axis) -> Self::Smaller {`
	`44`	`+ debug_assert!(axis.index() < self.ndim());`
`41`	`45`	`let mut tup = Dim([0; $n - 1]);`
`42`	`46`	`{`
`43`	`47`	`let mut it = tup.slice_mut().iter_mut();`