Skip to content

Commit b495d96

Browse files
committed
Add more standard dafusion namespaces to reduce clutter
1 parent 30b2543 commit b495d96

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

conftest.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,15 @@
2020
import datafusion as dfn
2121
import numpy as np
2222
import pytest
23+
from datafusion import col, lit
24+
from datafusion import functions as F
2325

2426

2527
@pytest.fixture(autouse=True)
2628
def _doctest_namespace(doctest_namespace: dict) -> None:
2729
"""Add common imports to the doctest namespace."""
2830
doctest_namespace["dfn"] = dfn
2931
doctest_namespace["np"] = np
32+
doctest_namespace["col"] = col
33+
doctest_namespace["lit"] = lit
34+
doctest_namespace["F"] = F

0 commit comments

Comments
 (0)