def test_decision_tree_max_depth(): model = tree.HoeffdingTreeClassifier() max_depths = [1, 2, 3, 4, 5, 6] models = utils.expand_param_grid(model, {"max_depth": max_depths}) for model, max_depth in zip(models, max_depths): assert model.max_depth == max_depth
def test_expand_param_grid_count(model, param_grid, count): assert len(utils.expand_param_grid(model, param_grid)) == count