Add some examples for plot implementations in their docstrings (#847)

Basically we finished the rough implementation of all plots (except 2 in DataFrame). So it should be better to have some examples for them.
databricks · Oct 2, 2019 · 89877ce · 89877ce
1 parent 1e96eb8
commit 89877ce
Show file tree

Hide file tree

Showing 3 changed files with 339 additions and 17 deletions.
diff --git a/databricks/conftest.py b/databricks/conftest.py
@@ -25,6 +25,7 @@
 
 import pandas as pd
 import pyarrow as pa
+import matplotlib.pyplot as plt
 from pyspark import __version__
 
 from databricks import koalas
@@ -85,3 +86,9 @@ def add_db(doctest_namespace):
 def add_caplog(caplog):
     with caplog.at_level(logging.INFO, logger='databricks.koalas.usage_logger'):
         yield
+
+
+@pytest.fixture(autouse=True)
+def close_figs():
+    yield
+    plt.close("all")