def test_integrated_dispersion_plot(self):
"""
Assert no errors occur during DispersionPlot integration
"""
text = [doc.split() for doc in corpus.data]
search_terms = ["Game", "player", "score", "oil", "Man"]
visualizer = DispersionPlot(search_terms)
visualizer.fit(text)
visualizer.ax.grid(False)
self.assert_images_similar(visualizer, tol=25)
def test_dispersion_plot_annotate_docs(self):
"""
Assert no errors occur during DispersionPlot integration
with annotate_docs parameter turned on
"""
text = [doc.split() for doc in corpus.data]
search_terms = ["girl", "she", "boy", "he", "man"]
visualizer = DispersionPlot(search_terms, annotate_docs=True)
visualizer.fit(text)
visualizer.ax.grid(False)
self.assert_images_similar(visualizer, tol=25.5)
def test_dispersion_plot_generator_input(self):
"""
Assert no errors occur during dispersionPlot integration
when the corpus' text type is a generator
"""
text = [doc.split() for doc in corpus.data]
search_terms = ["Game", "player", "score", "oil", "Man"]
visualizer = DispersionPlot(search_terms, ignore_case=True)
visualizer.fit(text)
visualizer.ax.grid(False)
self.assert_images_similar(visualizer, tol=25)
def test_dispersion_plot_ignore_case(self):
"""
Assert no errors occur during DispersionPlot integration
with ignore_case parameter turned on
"""
text = [doc.split() for doc in corpus.data]
target_words = ["Game", "player", "score", "oil", "Man"]
visualizer = DispersionPlot(target_words, ignore_case=True)
visualizer.fit(text)
visualizer.ax.grid(False)
self.assert_images_similar(visualizer, tol=25)
def test_dispersion_plot_color_by_class(self):
"""
Assert no errors occur during DispersionPlot integration
when target values are specified
"""
target = corpus.target
text = [doc.split() for doc in corpus.data]
search_terms = ["girl", "she", "boy", "he", "man"]
visualizer = DispersionPlot(search_terms)
visualizer.fit(text, target)
visualizer.ax.grid(False)
self.assert_images_similar(visualizer, tol=25)
def test_dispersion_plot_mismatched_labels(self):
"""
Assert exception is raised when number of labels doesn't match
"""
target = corpus.target
text = [doc.split() for doc in corpus.data]
search_terms = ["girl", "she", "boy", "he", "man"]
visualizer = DispersionPlot(search_terms,
annotate_docs=True,
labels=["a", "b"])
msg = (r"number of supplied labels \(\d\) "
r"does not match the number of classes \(\d\)")
with pytest.raises(YellowbrickValueError, match=msg):
visualizer.fit(text, target)
def dispersion(target_words, text, outpath, **kwargs):
# Create a new figure and axes
_, ax = plt.subplots()
# Visualize the Dispersion of target words
visualizer = DispersionPlot(target_words, ax=ax, **kwargs)
visualizer.fit(text)
visualizer.poof(outpath=outpath)