code: def test_return_type():
      """Check return type, should return an np.ndarray"""
      assert isinstance(data, np.ndarray)
  
imports: import numpy
  import pytest
  from impyute.dataset import mnist
  from impyute.ops import matrix
label_desc: Use image node_id 1 for calling a global function with example usage: test_return_type() without return types