Fix a bug that occurred when attempting to draw word clouds for empty (sub-)sets (issue #6)

Author: paulbrodersen

README.md

@@ -188,6 +188,13 @@ supervenn(sets, labels)
 plt.show()
 ```
 
+
+## Recent Changes
+
+- 0.1.1 Fix a bug that occurred when attempting to draw word clouds for empty (sub-)sets (issue #6)
+- 0.1.0 Implement set artists such that set outlines can be shown (issue #4)
+
+
 ## Contributing & Support
 
 If you get stuck and have a question that is not covered in the documentation, please raise an issue on the [issue tracker](https://github.com/paulbrodersen/matplotlib_set_diagrams/issues).

matplotlib_set_diagrams/__init__.py

@@ -33,7 +33,7 @@
 
 """
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 __author__ = "Paul Brodersen"
 __email__ = "paulbrodersen+matplotlib_set_diagrams@gmail.com"
 

matplotlib_set_diagrams/_diagram_classes.py

@@ -670,7 +670,7 @@ def _generate_wordcloud(
                 mask             = mask,
                 rgba             = self.subset_colors[subset_id],
                 wordcloud_kwargs = wordcloud_kwargs,
-            ) for subset_id, mask in subset_masks.items()
+            ) for subset_id, mask in subset_masks.items() if subsets[subset_id]
         ]
 
         combined_image = np.sum(subset_images, axis=0)

tests/test_diagram_classes.py

@@ -145,6 +145,17 @@ def test_EulerDiagram_as_wordcloud():
     return fig
 
 
+@pytest.mark.mpl_image_compare
+def test_EulerDiagram_as_wordcloud_with_empty_subsets():
+    sets = [
+        {"a", "b"},
+        {"c"}
+    ]
+    fig, ax = plt.subplots()
+    EulerDiagram.as_wordcloud(sets, wordcloud_kwargs=dict(random_state=42), ax=ax)
+    return fig
+
+
 @pytest.mark.mpl_image_compare
 def test_VennDiagram():
     fig, axes = plt.subplots(1, 2, sharex=True, sharey=True)