diff --git a/src/pyextremes/eva.py b/src/pyextremes/eva.py index d801076..59a696b 100644 --- a/src/pyextremes/eva.py +++ b/src/pyextremes/eva.py @@ -217,9 +217,12 @@ def AIC(self) -> float: def test_ks(self, significance_level: float = 0.05) -> KolmogorovSmirnov: return KolmogorovSmirnov( - extremes=self.extremes, + extremes=self.extremes_transformer.transformed_extremes, distribution=self.distribution.distribution, - fit_parameters=self.model.fit_parameters, + fit_parameters={ + **self.model.fit_parameters, + **self.model.distribution._fixed_parameters, + }, significance_level=significance_level, ) diff --git a/tests/integration/test_eva.py b/tests/integration/test_eva.py index 0730de2..44aaa31 100644 --- a/tests/integration/test_eva.py +++ b/tests/integration/test_eva.py @@ -592,7 +592,4 @@ def test_test_ks(self, eva_model_bm_mle, eva_model_pot_mle, extremes_method): "POT": eva_model_pot_mle, }[extremes_method] assert isinstance(eva_model.test_ks(), KolmogorovSmirnov) - if extremes_method == "BM": - assert eva_model.test_ks().success - else: - assert not eva_model.test_ks().success + assert eva_model.test_ks().success