Add a function `assert_polygons_grouped_by_type` like `assert_points_grouped_by_type` to ensure polygons are correctly plotted.