📝

tests/unit/test_plot.py

@@ -1,3 +1,7 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
+import pytest
 import matplotlib.pyplot as plt
 from reco_utils.common.plot import line_graph
 
@@ -16,9 +20,7 @@ def test_line_graph():
     plt.close()
 
     # Single graph as a subplot
-    line_graph(
-        values=[1, 2, 3], labels="Train", subplot=(1, 1, 1),
-    )
+    line_graph(values=[1, 2, 3], labels="Train", subplot=(1, 1, 1))
     plt.close()
 
     # Single graph with x values
@@ -29,3 +31,4 @@ def test_line_graph():
         y_min_max=(0, 5),
         plot_size=(5, 5),
     )
+    plt.close()