9
9
import pandas as pd
10
10
from pandas import DataFrame , Index , MultiIndex , Series , date_range
11
11
from pandas .core .computation .check import _NUMEXPR_INSTALLED
12
- import pandas .util . testing as tm
12
+ from pandas .tests . frame . common import TestData
13
13
from pandas .util .testing import (
14
14
assert_frame_equal ,
15
15
assert_series_equal ,
@@ -704,7 +704,7 @@ def setup_class(cls):
704
704
super ().setup_class ()
705
705
cls .engine = "numexpr"
706
706
cls .parser = "python"
707
- cls .frame = tm . makeDataFrame ()
707
+ cls .frame = TestData (). frame
708
708
709
709
def test_date_query_no_attribute_access (self ):
710
710
engine , parser = self .engine , self .parser
@@ -808,7 +808,7 @@ def setup_class(cls):
808
808
super ().setup_class ()
809
809
cls .engine = "python"
810
810
cls .parser = "pandas"
811
- cls .frame = tm . makeDataFrame ()
811
+ cls .frame = TestData (). frame
812
812
813
813
def test_query_builtin (self ):
814
814
engine , parser = self .engine , self .parser
@@ -827,7 +827,7 @@ class TestDataFrameQueryPythonPython(TestDataFrameQueryNumExprPython):
827
827
def setup_class (cls ):
828
828
super ().setup_class ()
829
829
cls .engine = cls .parser = "python"
830
- cls .frame = tm . makeDataFrame ()
830
+ cls .frame = TestData (). frame
831
831
832
832
def test_query_builtin (self ):
833
833
engine , parser = self .engine , self .parser
0 commit comments