Add legend to tests

a-maliarov · Apr 28, 2022 · 4945e5c · 4945e5c
1 parent 22aad2e
commit 4945e5c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/comparizon/mtest_dates_simpleplots.py b/comparizon/mtest_dates_simpleplots.py
@@ -27,10 +27,12 @@ def get_random_values(x, y):
     fig = Figure()
 
     xvalues, yvalues = get_random_values(1, 10000)
-    fig.plot(xvalues, yvalues, color='red', linewidth=5)
+    fig.plot(xvalues, yvalues, color='red', linewidth=5, label='line1')
 
     xvalues, yvalues = get_random_values(1, 10000)
-    fig.plot(xvalues, yvalues, color='blue', linewidth=5)
+    fig.plot(xvalues, yvalues, color='blue', linewidth=5, label='line1')
+
+    fig.legend()
 
     fig.save(f'{test_folder}/graph{i}.png')