def test__select_candidates_non_parametric(self):
# Run
candidates = Univariate._select_candidates(
parametric=ParametricType.NON_PARAMETRIC)
# Assert
assert candidates == [GaussianKDE]
def test__select_candidates_semibounded(self):
# Run
candidates = Univariate._select_candidates(
bounded=BoundedType.SEMI_BOUNDED)
# Assert
assert set(candidates) == {GammaUnivariate, LogLaplace}
def test__select_candidates_bounded(self):
# Run
candidates = Univariate._select_candidates(bounded=BoundedType.BOUNDED)
# Assert
assert set(candidates) == {
TruncatedGaussian, BetaUnivariate, UniformUnivariate
}
def test__select_candidates_unbounded(self):
# Run
candidates = Univariate._select_candidates(
bounded=BoundedType.UNBOUNDED)
# Assert
assert set(candidates) == {
GaussianKDE, GaussianUnivariate, StudentTUnivariate
}
def test__select_candidates(self):
# Run
candidates = Univariate._select_candidates()
# Assert
assert set(candidates) == {
GaussianKDE, GaussianUnivariate, TruncatedGaussian, BetaUnivariate,
GammaUnivariate, StudentTUnivariate, UniformUnivariate
}
def test__select_candidates_parametric(self):
# Run
candidates = Univariate._select_candidates(
parametric=ParametricType.PARAMETRIC)
# Assert
assert set(candidates) == {
GaussianUnivariate, TruncatedGaussian, BetaUnivariate,
GammaUnivariate, StudentTUnivariate, UniformUnivariate
}
