def test_chi_square_from_Dict2D(self):
"""chi_square_from_Dict2D calcs a Chi-Square and p value from Dict2D"""
# test1
obs_matrix = Dict2D({"rest_of_tree": {"env1": 2, "env3": 1, "env2": 0}, "b": {"env1": 1, "env3": 1, "env2": 3}})
input_matrix = calc_contingency_expected(obs_matrix)
test, csp = chi_square_from_Dict2D(input_matrix)
self.assertFloatEqual(test, 3.0222222222222221)
# test2
test_matrix_2 = Dict2D(
{
"Marl": {"val": [2, 5.2]},
"Chalk": {"val": [10, 5.2]},
"Sandstone": {"val": [8, 5.2]},
"Clay": {"val": [2, 5.2]},
"Limestone": {"val": [4, 5.2]},
}
)
test2, csp2 = chi_square_from_Dict2D(test_matrix_2)
self.assertFloatEqual(test2, 10.1538461538)
self.assertFloatEqual(csp2, 0.0379143890013)
# test3
matrix3_obs = Dict2D(
{"AIDS": {"Males": 4, "Females": 2, "Both": 3}, "No_AIDS": {"Males": 3, "Females": 16, "Both": 2}}
)
matrix3 = calc_contingency_expected(matrix3_obs)
test3, csp3 = chi_square_from_Dict2D(matrix3)
self.assertFloatEqual(test3, 7.6568405139833722)
self.assertFloatEqual(csp3, 0.0217439383468)
def test_calc_contingency_expected(self):
"""calcContingencyExpected returns new matrix with expected freqs"""
matrix = Dict2D({"rest_of_tree": {"env1": 2, "env3": 1, "env2": 0}, "b": {"env1": 1, "env3": 1, "env2": 3}})
result = calc_contingency_expected(matrix)
self.assertFloatEqual(result["rest_of_tree"]["env1"], [2, 1.125])
self.assertFloatEqual(result["rest_of_tree"]["env3"], [1, 0.75])
self.assertFloatEqual(result["rest_of_tree"]["env2"], [0, 1.125])
self.assertFloatEqual(result["b"]["env1"], [1, 1.875])
self.assertFloatEqual(result["b"]["env3"], [1, 1.25])
self.assertFloatEqual(result["b"]["env2"], [3, 1.875])
