add test

Author: hadia206

pydough/qdag/collections/range_collection.py

@@ -28,7 +28,12 @@ def __init__(
         end: int,
         step: int,
     ) -> None:
-        super().__init__(name=name, columns=[column_name])
+        super().__init__(
+            name=name,
+            columns=[
+                column_name,
+            ],
+        )
         self.start = start
         self.end = end
         self.step = step

pydough/qdag/collections/user_collection_qdag.py

@@ -35,8 +35,28 @@ def clone_with_parent(
         """
         return PyDoughUserGeneratedCollectionQDag(new_ancestor, self._collection)
 
+    def to_tree_form_isolated(self, is_last: bool) -> CollectionTreeForm:
+        # if self.ancestor_context is not None:
+        #     return CollectionTreeForm(
+        #         self.tree_item_string,
+        #         0,
+        #         has_predecessor=True,
+        #     )
+        # else:
+        return CollectionTreeForm(self.tree_item_string, 0)
+
     def to_tree_form(self, is_last: bool) -> CollectionTreeForm:
-        raise NotImplementedError
+        # if self.ancestor_context is not None:
+        #     predecessor: CollectionTreeForm = self.ancestor_context.to_tree_form(
+        #         is_last
+        #     )
+        #     predecessor.has_children = True
+        #     tree_form: CollectionTreeForm = self.to_tree_form_isolated(is_last)
+        #     tree_form.depth = predecessor.depth + 1
+        #     tree_form.predecessor = predecessor
+        #     return tree_form
+        # else:
+        return self.to_tree_form_isolated(is_last)
 
     @property
     def collection(self) -> PyDoughUserGeneratedCollection:
@@ -102,7 +122,7 @@ def standalone_string(self) -> str:
         Returns a string representation of the collection in a standalone form.
         This is used for debugging and logging purposes.
         """
-        return f"UserGeneratedCollection({self.name}, {', '.join(self.collection.columns)})"
+        return f"UserGeneratedCollection[{self.name}, {', '.join(self.collection.columns)}]"
 
     @property
     def key(self) -> str:
@@ -114,4 +134,4 @@ def to_string(self) -> str:
 
     @property
     def tree_item_string(self) -> str:
-        return f"UserGeneratedCollection({self.name})"
+        return f"UserGeneratedCollection[{self.name}: {', '.join(self.collection.columns)}]"

pydough/unqualified/unqualified_transform.py

@@ -360,8 +360,8 @@ def transform_code(
     source: str, graph_dict: dict[str, GraphMetadata], known_names: set[str]
 ) -> ast.AST:
     """
-    Transforms the source code into a new Python QDAG that has had the PyDough
-    decorator removed, had the definition of `_ROOT` injected at the top of the
+    Transforms the source code into a new Python QDAG that has the PyDough
+    decorator removed, has the definition of `_ROOT` injected at the top of the
     function body, and prepend unknown variables with `_ROOT.`
 
     Args:

pydough/user_collections/range_collection.py

@@ -5,10 +5,12 @@
 
 __all__ = ["range_collection"]
 
-from pydough.qdag.collections.range_collection import RangeGeneratedCollection
+from pydough.unqualified.unqualified_node import UnqualifiedGeneratedCollection
 
 
-def range_collection(name: str, column: str, *args: int) -> RangeGeneratedCollection:
+def range_collection(
+    name: str, column: str, *args: int
+) -> UnqualifiedGeneratedCollection:
     """
     Creates a collection of integer ranges over a specified column.
 
@@ -63,10 +65,18 @@ def range_collection(name: str, column: str, *args: int) -> RangeGeneratedCollec
     if step <= 0:
         raise ValueError(f"Expected 'step' to be a positive integer, got {step}")
 
-    return RangeGeneratedCollection(
-        name=name,
-        column_name=column,
-        start=start,
-        end=end,
-        step=step,
+    # return RangeGeneratedCollection(
+    #     name=name,
+    #     column_name=column,
+    #     start=start,
+    #     end=end,
+    #     step=step,
+    # )
+    range_args = [start, end, step]
+    return UnqualifiedGeneratedCollection(
+        name,
+        [
+            column,
+        ],
+        range_args,
     )

tests/test_pydough_functions/user_collections.py

@@ -6,13 +6,18 @@
 # mypy: ignore-errors
 # ruff & mypy should not try to typecheck or verify any of this
 
-import pandas as pd
-import datetime
-
 import pydough
 
 
-def simple_range():
+def simple_range_1():
+    return pydough.range_collection(
+        "simple_range",
+        "value",
+        10,  # end value
+    )
+
+
+def simple_range_2():
     return pydough.range_collection(
         "simple_range",
         "value",

tests/test_qualification.py

@@ -64,7 +64,8 @@
     impl_tpch_q22,
 )
 from tests.test_pydough_functions.user_collections import (
-    simple_range,
+    simple_range_1,
+    simple_range_2,
 )
 from tests.testing_utilities import (
     graph_fetcher,
@@ -946,10 +947,17 @@
             id="simple_cross_6",
         ),
         pytest.param(
-            simple_range,
+            simple_range_1,
             """
+  UserGeneratedCollection[simple_range: value]
             """,
-            id="simple_range",
+            id="simple_range_1",
+        ),
+        pytest.param(
+            simple_range_2,
+            """
+            """,
+            id="simple_range_2",
         ),
     ],
 )