def test__group_data(self):
mrbbagging = MRBBagging(1, tree.DecisionTreeClassifier())
x = [[1, 1, 1], [2, 2, 2], [3, 3, 3]]
y = ["A", "B", "C"]
classes, grouped_data = mrbbagging._group_data(x, y)
self.assertEqual(classes, {'A', 'B', 'C'})
self.assertEqual(
grouped_data, {
'C': [[[3, 3, 3], 'C']],
'A': [[[1, 1, 1], 'A']],
'B': [[[2, 2, 2], 'B']]
})
def test__group_data_with_none(self):
mrbbagging = MRBBagging(1, tree.DecisionTreeClassifier())
x = [[1, 1, 1], [2, 2, 2], [3, 3, 3]]
y = ["A", None, "C"]
with self.assertRaises(AssertionError):
mrbbagging._group_data(x, y)