✅ added tests for lineplot

tests/conftest.py

@@ -3,7 +3,7 @@
 import pandas as pd
 import pytest
 
-from pynimate.datafier import BarDatafier, BaseDatafier
+from pynimate.datafier import BarDatafier, BaseDatafier, LineDatafier
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
 
@@ -43,6 +43,11 @@ def sample_data1_bardfr(sample_data1) -> BarDatafier:
     return BarDatafier(sample_data1, "%Y-%m-%d", "3MS")
 
 
+@pytest.fixture
+def sample_data1_linedfr(sample_data1) -> BarDatafier:
+    return LineDatafier(sample_data1, "%Y-%m-%d", "3MS")
+
+
 @pytest.fixture
 def map_data() -> pd.DataFrame:
     map_data = pd.read_csv(dir_path + "/data/map.csv").set_index("time")

tests/test_datafier.py

@@ -147,6 +147,9 @@ def test_datafier_get_prepared_data(sample_data1):
     assert dfr.data.equals(data)
 
 
+# 1.2.0 >=
+
+
 def test_basedatafier_interpolate_even(sample_data2):
     dfr = BaseDatafier(sample_data2, "%Y", "3MS")
     interpolated_data = pd.DataFrame(

tests/test_lineplot.py

@@ -0,0 +1,28 @@
+from pynimate.lineplot import Lineplot
+
+
+def test_lineplot_linestyle(sample_data1_linedfr):
+    plot = Lineplot(sample_data1_linedfr)
+    linestyles = {
+        "Afghanistan": "solid",
+        "Angola": "solid",
+        "Albania": "solid",
+        "USA": "solid",
+        "Argentina": "solid",
+    }
+
+    assert plot.column_linestyles == linestyles
+
+
+def test_lineplot_linestyle_dict(sample_data1_linedfr):
+    plot = Lineplot(sample_data1_linedfr)
+    plot.set_column_linestyles({"Albania": "dashed"})
+    linestyles = {
+        "Afghanistan": "solid",
+        "Angola": "solid",
+        "Albania": "dashed",
+        "USA": "solid",
+        "Argentina": "solid",
+    }
+
+    assert plot.column_linestyles == linestyles