Add some tests and remove .coveragerc

bayesclass/tests/test_AODE.py

@@ -2,6 +2,8 @@ import pytest
 import numpy as np
 from sklearn.datasets import load_iris
 from sklearn.preprocessing import KBinsDiscretizer
+from matplotlib.testing.decorators import image_comparison
+from matplotlib.testing.conftest import mpl_test_settings
 
 
 from bayesclass.bayesclass import AODE
@@ -37,6 +39,17 @@ def test_AODE_default_hyperparameters(data, clf):
     ]
 
 
+@image_comparison(
+    baseline_images=["line_dashes_AODE"], remove_text=True, extensions=["png"]
+)
+def test_AODE_plot(data, clf):
+    # mpl_test_settings will automatically clean these internal side effects
+    mpl_test_settings
+    dataset = load_iris(as_frame=True)
+    clf.fit(*data, features=dataset["feature_names"])
+    clf.plot("AODE Iris")
+
+
 def test_AODE_version(clf):
     """Check AODE version."""
     assert __version__ == clf.version()

bayesclass/tests/test_doctest.py

@@ -1,7 +0,0 @@
-import doctest
-import bayesclass
-
-
-def load_tests(loader, tests, ignore):
-    tests.addTests(doctest.DocTestSuite(bayesclass))
-    return tests