Skip to content

Commit 04661ac

Browse files
committed
fix: summarize all the unit tests in test_pit.
1 parent 90e9d8a commit 04661ac

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

tests/test_pit.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ def to_str(self, obj):
7474
def check_same(self, a, b):
7575
self.assertEqual(self.to_str(a), self.to_str(b))
7676

77+
@pytest.mark.slow
7778
def test_query(self):
7879
instruments = ["sh600519"]
7980
fields = ["P($$roewa_q)", "P($$yoyni_q)"]
@@ -104,6 +105,7 @@ def test_query(self):
104105
"""
105106
self.check_same(data.tail(), res)
106107

108+
@pytest.mark.slow
107109
def test_no_exist_data(self):
108110
fields = ["P($$roewa_q)", "P($$yoyni_q)", "$close"]
109111
data = D.features(["sh600519", "sh601988"], fields, start_time="2019-01-01", end_time="2019-07-19", freq="day")
@@ -159,6 +161,7 @@ def test_expr(self):
159161
"""
160162
self.check_same(data.tail(15), expect)
161163

164+
@pytest.mark.slow
162165
def test_unlimit(self):
163166
# fields = ["P(Mean($$roewa_q, 1))", "P($$roewa_q)", "P(Mean($$roewa_q, 2))", "P(Ref($$roewa_q, 1))", "P((Ref($$roewa_q, 1) +$$roewa_q) / 2)"]
164167
fields = ["P($$roewa_q)"]
@@ -227,6 +230,7 @@ def test_unlimit(self):
227230
"""
228231
self.check_same(s[~s.duplicated().values], expect)
229232

233+
@pytest.mark.slow
230234
def test_expr2(self):
231235
instruments = ["sh600519"]
232236
fields = ["P($$roewa_q)", "P($$yoyni_q)"]
@@ -253,6 +257,7 @@ def test_expr2(self):
253257
"""
254258
self.check_same(data, except_data)
255259

260+
@pytest.mark.slow
256261
def test_pref_operator(self):
257262
instruments = ["sh600519"]
258263
fields = [

0 commit comments

Comments
 (0)