old_contents: # -*- coding: utf-8 -*-
  
new_contents: # -*- coding: utf-8 -*-
  
  
  import pytest
  
  import numpy as np
  
  import dask.array as da
  import dask.array.utils as dau
  
  import dask_ndfourier._utils
  
  
  @pytest.mark.parametrize(
      "a, s, n, axis", [
          (da.ones((3, 4), chunks=(3, 4)), da.ones((2,), chunks=(2,)), -1, -1),
      ]
  )
  def test_norm_args(a, s, n, axis):
      s2, n2, axis2 = dask_ndfourier._utils._norm_args(a, s, n=n, axis=axis)
  
      assert isinstance(s2, da.Array)
  
subject: Test that `s` as a Dask Array is preserved