def test_3d_shapes():
 w = wcs.WCS(naxis=3)
 with NumpyRNGContext(123456789):
  data = np.random.rand(100, 3)
  result = w.wcs_pix2world(data, 1)
  assert (result.shape == (100, 3))
  result = w.wcs_pix2world(data[..., 0], data[..., 1], data[..., 2], 1)
  assert (len(result) == 3)

def test_3d_shapes ( ) : w = wcs . WCS ( naxis = 3 ) with NumpyRNGContext ( 123456789 ) : data = np . random . rand ( 100 , 3 ) result = w . wcs_pix2world ( data , 1 ) assert ( result . shape == ( 100 , 3 ) ) result = w . wcs_pix2world ( data [ ... , 0 ] , data [ ... , 1 ] , data [ ... , 2 ] , 1 ) assert ( len ( result ) == 3 )
issue #444 .