def test_get_grouped_stats_returns_correct_number_of_statistics():
num_statistics = 5
statistics = [_DummyStatistic("c" + str(i), False)
for i in range(num_statistics)]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
assert len(stats.columns) == num_statistics
def test_get_grouped_stats_returns_correct_column_names():
name1 = "dummy1"
name2 = "dummy2"
statistics = [_DummyStatistic(name1, False), _DummyStatistic(name2, False)]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
assert name1 in stats.columns
assert name2 in stats.columns
def test_get_grouped_stats_returns_correct_column_names():
name1 = "dummy1"
name2 = "dummy2"
statistics = [_DummyStatistic(name1, False), _DummyStatistic(name2, False)]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
assert name1 in stats.columns
assert name2 in stats.columns
def test_get_grouped_stats_returns_correct_number_of_statistics():
num_statistics = 5
statistics = [
_DummyStatistic("c" + str(i), False) for i in range(num_statistics)
]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
assert len(stats.columns) == num_statistics
def test_get_grouped_stats_calculates_correct_values():
name1 = "dummy1"
name2 = "dummy2"
statistics = [_DummyStatistic(name1, False), _DummyStatistic(name2, True)]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
for group in set(GROUPS):
assert stats[name1].ix[group] == len(tpms[tpms[GROUP_TEST_COL] == group])
assert stats[name2].ix[group] == len(tp_tpms[tp_tpms[GROUP_TEST_COL] == group])
def test_get_grouped_stats_calculates_correct_values():
name1 = "dummy1"
name2 = "dummy2"
statistics = [_DummyStatistic(name1, False), _DummyStatistic(name2, True)]
tpms = _get_test_tpms()
tp_tpms = _get_test_tp_tpms()
stats = t.get_grouped_stats(tpms, tp_tpms, GROUP_TEST_COL, statistics)
for group in set(GROUPS):
assert stats[name1].ix[group] == len(
tpms[tpms[GROUP_TEST_COL] == group])
assert stats[name2].ix[group] == len(
tp_tpms[tp_tpms[GROUP_TEST_COL] == group])
