def assert_newaxis_keys(self, shape, new_axes, expected): in_arr = ConstantArray(shape) array = NewAxesArray(in_arr, new_axes) keys = array._newaxis_keys() self.assertEqual(keys, expected)
def assert_newaxis_keys(self, shape, new_axes, expected): in_arr = ConstantArray(shape) array = NewAxesArray(in_arr, new_axes) keys = array._newaxis_keys() self.assertEqual(keys, expected)