Skip to content

Commit e32117b

Browse files
committed
Added to string test
1 parent 28b3f5c commit e32117b

File tree

2 files changed

+27
-3
lines changed

2 files changed

+27
-3
lines changed

pydough/relational/root.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(
2525
self,
2626
input: Relational,
2727
columns: MutableSequence["Column"],
28-
orderings: MutableSequence["PyDoughExpressionAST"] | None,
28+
orderings: MutableSequence["PyDoughExpressionAST"] | None = None,
2929
) -> None:
3030
super().__init__(input, columns, orderings)
3131

tests/test_relational.py

Lines changed: 26 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from pydough.relational.filter import Filter
2020
from pydough.relational.limit import Limit
2121
from pydough.relational.project import Project
22+
from pydough.relational.root import RelationalRoot
2223
from pydough.relational.scan import Scan
2324
from pydough.types import BooleanType, Int64Type
2425

@@ -2295,8 +2296,31 @@ def test_filter_invalid_merge(first_filter: Filter, second_filter: Filter):
22952296
first_filter.merge(second_filter)
22962297

22972298

2298-
# def test_root_to_string(root: RelationalRoot, output: str):
2299-
# assert root.to_string() == output
2299+
@pytest.mark.parametrize(
2300+
"root, output",
2301+
[
2302+
pytest.param(
2303+
RelationalRoot(
2304+
build_simple_scan(),
2305+
[make_column("a"), make_column("b")],
2306+
),
2307+
"ROOT(columns=[Column(name='a', expr=Column(a)), Column(name='b', expr=Column(b))], orderings=[])",
2308+
id="no_orderings",
2309+
),
2310+
pytest.param(
2311+
RelationalRoot(
2312+
build_simple_scan(),
2313+
[make_column("a"), make_column("b")],
2314+
[make_simple_column_reference("a")],
2315+
),
2316+
"ROOT(columns=[Column(name='a', expr=Column(a)), Column(name='b', expr=Column(b))], orderings=[Column(a)])",
2317+
id="with_orderings",
2318+
),
2319+
],
2320+
)
2321+
def test_root_to_string(root: RelationalRoot, output: str):
2322+
assert root.to_string() == output
2323+
23002324

23012325
# def test_root_equals(first_root: RelationalRoot, second_root: Relational, output: bool):
23022326
# assert first_root.equals(second_root) == output

0 commit comments

Comments
 (0)