## Licensed to the Apache Software Foundation (ASF) under one or more# contributor license agreements. See the NOTICE file distributed with# this work for additional information regarding copyright ownership.# The ASF licenses this file to You under the Apache License, Version 2.0# (the "License"); you may not use this file except in compliance with# the License. You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License.#fromtypingimportGeneric,Tuple,TypeVarfrompyspark.mllib.commonimportinherit_doc,JavaModelWrapper__all__=["ChiSqTestResult","KolmogorovSmirnovTestResult"]DF=TypeVar("DF",int,float,Tuple[int,...],Tuple[float,...])classTestResult(JavaModelWrapper,Generic[DF]):""" Base class for all test results. """@propertydefpValue(self)->float:""" The probability of obtaining a test statistic result at least as extreme as the one that was actually observed, assuming that the null hypothesis is true. """returnself._java_model.pValue()@propertydefdegreesOfFreedom(self)->DF:""" Returns the degree(s) of freedom of the hypothesis test. Return type should be Number(e.g. Int, Double) or tuples of Numbers. """returnself._java_model.degreesOfFreedom()@propertydefstatistic(self)->float:""" Test statistic. """returnself._java_model.statistic()@propertydefnullHypothesis(self)->str:""" Null hypothesis of the test. """returnself._java_model.nullHypothesis()def__str__(self)->str:returnself._java_model.toString()
[docs]@inherit_docclassChiSqTestResult(TestResult[int]):""" Contains test results for the chi-squared hypothesis test. """@propertydefmethod(self)->str:""" Name of the test method """returnself._java_model.method()
[docs]@inherit_docclassKolmogorovSmirnovTestResult(TestResult[int]):""" Contains test results for the Kolmogorov-Smirnov test. """