old_contents: """
  Test return values
  """
  
  from __future__ import print_function
  
  import numba.unittest_support as unittest
  from numba.compiler import compile_isolated, Flags
  from numba.utils import PYVERSION
  from numba import types
  from numba.typeinfer import TypingError
  import math
  
  
  enable_pyobj_flags = Flags()
  enable_pyobj_flags.set("enable_pyobject")
  no_pyobj_flags = Flags()
  
  
  def get_nopython_func():
      return abs
  
  def get_pyobj_func():
      return pow
  
  def get_module_func():
      return math.floor
  
  
  class TestReturnValues(unittest.TestCase):
  
      def test_nopython_func(self, flags=enable_pyobj_flags):
          # Test returning func that is supported in nopython mode
          pyfunc = get_nopython_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, abs)
          else:
              result = cfunc()
  
      def test_nopython_func_npm(self):
          with self.assertRaises(TypeError):
              self.test_nopython_func(flags=no_pyobj_flags)
  
      def test_pyobj_func(self, flags=enable_pyobj_flags):
          # Test returning func that is only supported in object mode
          pyfunc = get_pyobj_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, pow)
          else:
              result = cfunc()
  
      def test_pyobj_func_npm(self):
          with self.assertRaises(TypeError):
              self.test_pyobj_func(flags=no_pyobj_flags)
  
      def test_module_func(self, flags=enable_pyobj_flags):
          # Test returning imported func that is only supported in object mode
          pyfunc = get_module_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, math.floor)
          else:
              result = cfunc()
  
      def test_module_func_npm(self):
          with self.assertRaises(TypeError):
              self.test_module_func(flags=no_pyobj_flags)
  
  
  if __name__ == '__main__':
      unittest.main()
  
new_contents: """
  Test return values
  """
  
  from __future__ import print_function
  
  import numba.unittest_support as unittest
  from numba.compiler import compile_isolated, Flags
  from numba.utils import PYVERSION
  from numba import types
  from numba.typeinfer import TypingError
  import math
  
  
  enable_pyobj_flags = Flags()
  enable_pyobj_flags.set("enable_pyobject")
  no_pyobj_flags = Flags()
  
  
  def get_nopython_func():
      return abs
  
  def get_pyobj_func():
      return open
  
  def get_module_func():
      return math.floor
  
  
  class TestReturnValues(unittest.TestCase):
  
      def test_nopython_func(self, flags=enable_pyobj_flags):
          # Test returning func that is supported in nopython mode
          pyfunc = get_nopython_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, abs)
          else:
              result = cfunc()
  
      def test_nopython_func_npm(self):
          with self.assertRaises(TypeError):
              self.test_nopython_func(flags=no_pyobj_flags)
  
      def test_pyobj_func(self, flags=enable_pyobj_flags):
          # Test returning func that is only supported in object mode
          pyfunc = get_pyobj_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, open)
          else:
              result = cfunc()
  
      def test_pyobj_func_npm(self):
          with self.assertRaises(TypingError):
              self.test_pyobj_func(flags=no_pyobj_flags)
  
      def test_module_func(self, flags=enable_pyobj_flags):
          # Test returning imported func that is only supported in object mode
          pyfunc = get_module_func
          cr = compile_isolated(pyfunc, (), flags=flags)
          cfunc = cr.entry_point
          if flags == enable_pyobj_flags:
              result = cfunc()
              self.assertEqual(result, math.floor)
          else:
              result = cfunc()
  
      def test_module_func_npm(self):
          with self.assertRaises(TypeError):
              self.test_module_func(flags=no_pyobj_flags)
  
  
  if __name__ == '__main__':
      unittest.main()
  
subject: Change test_return_values test to return 'open' function instead of 'pow'