code: def test_wrightomega_nan():
      pts = [
          complex(np.nan, 0),
          complex(0, np.nan),
          complex(np.nan, np.nan),
          complex(np.nan, 1),
          complex(1, np.nan),
      ]
      for p in pts:
          res = sc.wrightomega(p)
          assert_(np.isnan(res.real))
          assert_(np.isnan(res.imag))
  
file_path: scipy/scipy/special/tests/test_wrightomega.py
imports: import pytest
  import numpy
  from numpy.testing import assert_, assert_equal, assert_allclose
  import scipy.special
  from scipy.special._testutils import assert_func_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_wrightomega_nan() without return types