From 705bc6ec82001ec0908d268f45ac4aefe88d6209 Mon Sep 17 00:00:00 2001
From: David de la Iglesia Castro <daviddelaiglesiacastro@gmail.com>
Date: Sat, 19 Nov 2022 10:24:47 +0100
Subject: [PATCH] Added test_infer_x_label

---
 tests/unit/render/test_vega_converter.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/tests/unit/render/test_vega_converter.py b/tests/unit/render/test_vega_converter.py
index 6a99b5d96..7f3c85479 100644
--- a/tests/unit/render/test_vega_converter.py
+++ b/tests/unit/render/test_vega_converter.py
@@ -248,3 +248,17 @@ def test_convert(
 
     assert datapoints == expected_datapoints
     assert resolved_properties == expected_properties
+
+
+@pytest.mark.parametrize(
+    "properties,label",
+    [
+        ({"x": {"actual.csv": "actual"}}, "actual"),
+        (
+            {"x": {"train_actual.csv": "actual", "val_actual.csv": "actual"}},
+            "actual",
+        ),
+    ],
+)
+def test_infer_x_label(properties, label):
+    assert VegaConverter.infer_x_label(properties) == label