code: def test_non_default_kpss():
      test = KPSSTest(alpha=0.05, null="trend", lshort=False)
      pval, do_diff = test.should_diff(austres)
      assert do_diff  # show it is significant
      assert np.allclose(pval, 0.01, atol=0.005)
  
      # test the ndiffs with the KPSS test
      assert ndiffs(austres, test="kpss", max_d=2) == 2
  
file_path: pmdarima/pmdarima/arima/tests/test_stationarity.py
imports: from pmdarima.arima.stationarity import ADFTest, PPTest, KPSSTest
  from pmdarima.arima.utils import ndiffs
  from pmdarima.utils.array import diff
  from pmdarima.datasets import load_austres
  from sklearn.utils import check_random_state
  from numpy.testing import assert_array_almost_equal, assert_almost_equal, assert_array_equal
  import numpy
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: test_non_default_kpss() without return types