Обновлены тесты

Author: SergeyShk

tests/visualizers/test_fingerprinting.py

@@ -19,7 +19,7 @@ def test_fingerprinting_type_error():
 def test_fingerprinting(texts):
     plt.cla()
     plot = fingerprinting(texts, x_size=600, y_size=500)
-    assert type(plot) == Figure
+    assert isinstance(plot, Figure)
     assert len(plot.axes) == 2
     assert plot.axes[0].title.get_text() == "Литературная дактилоскопия"
     assert plot.axes[0].title.get_position() == (0.5, 1.0)
@@ -37,4 +37,4 @@ def test_fingerprinting_is_return(texts):
 def test_fingerprinting_metric(texts):
     plt.cla()
     plot = fingerprinting(texts, metric=calc_simpson_index)
-    assert type(plot) == Figure
+    assert isinstance(plot, Figure)

tests/visualizers/test_word_tree.py

@@ -11,7 +11,7 @@ def texts():
 
 def test_wordtree_type_error():
     with pytest.raises(TypeError):
-        wordtree(1)
+        wordtree(1, "тест")
 
 
 def test_wordtree_value_error(texts):
@@ -21,6 +21,6 @@ def test_wordtree_value_error(texts):
 
 def test_wordtree(texts):
     g = wordtree(texts, "рабочий", max_n=2)
-    assert type(g) == Digraph
+    assert isinstance(g, Digraph)
     assert len(g.body) == 11
     assert g.name == "рабочий"

tests/visualizers/test_zipf.py

@@ -15,7 +15,7 @@ def tokens():
 def test_zipf_theory():
     plt.cla()
     plot = zipf_theory(10, 5, 1.0)[0]
-    assert type(plot) == Line2D
+    assert isinstance(plot, Line2D)
     assert plot._label == "Теоретический закон"
     assert plot.get_linewidth() == 2
     assert plot.get_color() == "r"
@@ -29,7 +29,7 @@ def test_zipf_type_error():
 def test_zipf(tokens):
     plt.cla()
     plot = zipf(tokens)
-    assert type(plot) == Line2D
+    assert isinstance(plot, Line2D)
     assert plot._label == "Экспериментальный закон"
     assert plot.axes.title.get_text() == "Закон Ципфа"
     assert plot.axes.title.get_position() == (0.5, 1.0)
@@ -66,4 +66,4 @@ def test_zipf_with_theory(tokens):
     plt.cla()
     plot = zipf(tokens, show_theory=True)
     assert len(plot) == 1
-    assert all(type(p) == Line2D for p in plot)
+    assert all(isinstance(p, Line2D) for p in plot)