Added test_infer_x_label

2026-11-06 07:18:28 +01:00 · 2022-11-21 21:05:12 +01:00
parent 61a02e6f56
commit 705bc6ec82
1 changed files with 14 additions and 0 deletions
@@ -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