t.vect.db.select: Add pytest files

temporal/t.vect.db.select/tests/conftest.py

@@ -0,0 +1,88 @@
+"""Fixture for t.vect.db.select tests"""
+
+import os
+from types import SimpleNamespace
+
+import pytest
+
+import grass.script as gs
+from grass.tools import Tools
+
+
+@pytest.fixture(scope="module")
+def space_time_vector_dataset(tmp_path_factory):
+    """Start a temporary session and create space time vector datasets.
+
+    Returns an object containing metadata attributes about the generated datasets.
+    """
+    tmp_path = tmp_path_factory.mktemp("vector_time_series")
+    project = tmp_path / "test_project_vector"
+    gs.create_project(project)
+
+    with gs.setup.init(project, env=os.environ.copy()) as session:
+        tools = Tools(session=session)
+
+        tools.g_region(s=0, n=80, w=0, e=120, b=0, t=50, res=10, res3=10)
+
+        for i in range(1, 4):
+            tools.r_mapcalc(expression=f"prec_int_{i} = {i}00.0", overwrite=True)
+            tools.r_mapcalc(expression=f"prec_inst_{i} = {i}00.0", overwrite=True)
+
+        tools.v_random(output="prec_points", n=3, seed=1, flags="z", overwrite=True)
+
+        interval_strds = "precip_abs1"
+        tools.t_create(
+            type="strds",
+            temporaltype="absolute",
+            output=interval_strds,
+            title="A test",
+            description="A test",
+        )
+        tools.t_register(
+            type="raster",
+            flags="i",
+            input=interval_strds,
+            maps=",".join([f"prec_int_{i}" for i in range(1, 4)]),
+            start="2001-03-01 00:00:00",
+            increment="1 months",
+        )
+
+        interval_stvds = "prec_observer_interval"
+        tools.t_vect_observe_strds(
+            input="prec_points",
+            strds=interval_strds,
+            output=interval_stvds,
+            vector=interval_stvds,
+            column="observation",
+        )
+
+        instance_strds = "precip_abs2"
+        tools.t_create(
+            type="strds",
+            temporaltype="absolute",
+            output=instance_strds,
+            title="B test",
+            description="B test",
+        )
+        tools.t_register(
+            type="raster",
+            input=instance_strds,
+            maps=",".join([f"prec_inst_{i}" for i in range(1, 4)]),
+            start="2004-01-01 00:00:00",
+            increment="3 months",
+        )
+
+        instance_stvds = "prec_observer_instance"
+        tools.t_vect_observe_strds(
+            input="prec_points",
+            strds=instance_strds,
+            output=instance_stvds,
+            vector=instance_stvds,
+            column="observation",
+        )
+
+        yield SimpleNamespace(
+            session=session,
+            interval_name=interval_stvds,
+            instance_name=instance_stvds,
+        )

temporal/t.vect.db.select/tests/t_vect_db_select_test.py

@@ -0,0 +1,123 @@
+"""Test t.vect.db.select functionality against expected outputs."""
+
+from grass.tools import Tools
+
+
+def test_t_vect_db_select_interval_default(space_time_vector_dataset):
+    """Verify standard plain text layout for interval datasets."""
+    tools = Tools(session=space_time_vector_dataset.session)
+
+    result = tools.t_vect_db_select(input=space_time_vector_dataset.interval_name)
+    assert result.returncode == 0
+
+    expected_output = """start_time|end_time|cat|observation
+2001-03-01 00:00:00|2001-04-01 00:00:00|1|100
+2001-03-01 00:00:00|2001-04-01 00:00:00|2|100
+2001-03-01 00:00:00|2001-04-01 00:00:00|3|100
+2001-04-01 00:00:00|2001-05-01 00:00:00|1|200
+2001-04-01 00:00:00|2001-05-01 00:00:00|2|200
+2001-04-01 00:00:00|2001-05-01 00:00:00|3|200
+2001-05-01 00:00:00|2001-06-01 00:00:00|1|300
+2001-05-01 00:00:00|2001-06-01 00:00:00|2|300
+2001-05-01 00:00:00|2001-06-01 00:00:00|3|300"""
+
+    actual_lines = result.text_split("\n")
+    expected_lines = expected_output.strip().splitlines()
+
+    for actual, expected in zip(actual_lines, expected_lines, strict=True):
+        assert actual == expected
+
+
+def test_t_vect_db_select_interval_where(space_time_vector_dataset):
+    """Verify filtering via attribute clauses with custom separators on intervals."""
+    tools = Tools(session=space_time_vector_dataset.session)
+
+    result = tools.t_vect_db_select(
+        input=space_time_vector_dataset.interval_name,
+        where="cat = 1",
+        separator="  |  ",
+    )
+    assert result.returncode == 0
+
+    expected_output = """start_time  |  end_time  |  cat  |  observation
+2001-03-01 00:00:00  |  2001-04-01 00:00:00  |  1  |  100
+2001-04-01 00:00:00  |  2001-05-01 00:00:00  |  1  |  200
+2001-05-01 00:00:00  |  2001-06-01 00:00:00  |  1  |  300"""
+
+    actual_lines = result.text_split("\n")
+    expected_lines = expected_output.strip().splitlines()
+
+    for actual, expected in zip(actual_lines, expected_lines, strict=True):
+        assert actual == expected
+
+
+def test_t_vect_db_select_interval_columns(space_time_vector_dataset):
+    """Verify selective attribute column specification on intervals."""
+    tools = Tools(session=space_time_vector_dataset.session)
+
+    result = tools.t_vect_db_select(
+        input=space_time_vector_dataset.interval_name,
+        where="cat = 1",
+        separator="  |  ",
+        columns="observation",
+    )
+    assert result.returncode == 0
+
+    expected_output = """start_time  |  end_time  |  observation
+2001-03-01 00:00:00  |  2001-04-01 00:00:00  |  100
+2001-04-01 00:00:00  |  2001-05-01 00:00:00  |  200
+2001-05-01 00:00:00  |  2001-06-01 00:00:00  |  300"""
+
+    actual_lines = result.text_split("\n")
+    expected_lines = expected_output.strip().splitlines()
+
+    for actual, expected in zip(actual_lines, expected_lines, strict=True):
+        assert actual == expected
+
+
+def test_t_vect_db_select_instance_default(space_time_vector_dataset):
+    """Verify standard plain text layout for instance datasets."""
+    tools = Tools(session=space_time_vector_dataset.session)
+
+    result = tools.t_vect_db_select(input=space_time_vector_dataset.instance_name)
+    assert result.returncode == 0
+
+    expected_output = """start_time|end_time|cat|observation
+2004-01-01 00:00:00||1|100
+2004-01-01 00:00:00||2|100
+2004-01-01 00:00:00||3|100
+2004-04-01 00:00:00||1|200
+2004-04-01 00:00:00||2|200
+2004-04-01 00:00:00||3|200
+2004-07-01 00:00:00||1|300
+2004-07-01 00:00:00||2|300
+2004-07-01 00:00:00||3|300"""
+
+    actual_lines = result.text_split("\n")
+    expected_lines = expected_output.strip().splitlines()
+
+    for actual, expected in zip(actual_lines, expected_lines, strict=True):
+        assert actual == expected
+
+
+def test_t_vect_db_select_instance_where(space_time_vector_dataset):
+    """Verify filtering via attribute clauses with custom separators on instances."""
+    tools = Tools(session=space_time_vector_dataset.session)
+
+    result = tools.t_vect_db_select(
+        input=space_time_vector_dataset.instance_name,
+        where="cat = 1",
+        separator="  |  ",
+    )
+    assert result.returncode == 0
+
+    expected_output = """start_time  |  end_time  |  cat  |  observation
+2004-01-01 00:00:00  |    |  1  |  100
+2004-04-01 00:00:00  |    |  1  |  200
+2004-07-01 00:00:00  |    |  1  |  300"""
+
+    actual_lines = result.text_split("\n")
+    expected_lines = expected_output.strip().splitlines()
+
+    for actual, expected in zip(actual_lines, expected_lines, strict=True):
+        assert actual == expected