Commit c5e0d11c authored by Levin Zimmermann's avatar Levin Zimmermann

erp5_core_test: restricted: Test pandas resampler

In some projects we use pandas resampler inside 'portal_callables'
(so with restricted Python). But we miss tests which check
whether pandas resamplers are allowed inside the restricted Python.
Due to this lack of tests we won't notice whether the usage of pandas
resamplers still work with a newer pandas version. With this patch
we can recognize if an internal change of pandas makes the resampler unusuable
in restricted Python.

/reviewed-on !1738
/reviewed-by @jerome @klaus
parent b88ea7b1
...@@ -572,6 +572,39 @@ class TestRestrictedPythonSecurity(ERP5TypeTestCase): ...@@ -572,6 +572,39 @@ class TestRestrictedPythonSecurity(ERP5TypeTestCase):
expected=999 expected=999
) )
def testPandasDatetimeIndexResampler(self):
self.createAndRunScript('''
import pandas as pd
index = pd.date_range('1/1/2000', periods=9, freq='T')
series = pd.Series(range(9), index=index)
resampler = series.resample('3T')
return resampler.mean()[0]
''',
expected=1
)
def testPandasPeriodIndexResampler(self):
self.createAndRunScript('''
import pandas as pd
index = pd.period_range(start='2017-01-01', end='2018-01-01', freq='M')
series = pd.Series(range(len(index)), index=index)
resampler = series.resample('3T')
return resampler.mean()[0]
''',
expected=0.0
)
def testPandasTimedeltaIndexResampler(self):
self.createAndRunScript('''
import pandas as pd
index = pd.timedelta_range(start='1 day', periods=4)
series = pd.Series(range(len(index)), index=index)
resampler = series.resample('3T')
return resampler.mean()[0]
''',
expected=0.0
)
def testPandasIORead(self): def testPandasIORead(self):
# Test the black_list configuration validity # Test the black_list configuration validity
for read_method in pandas_black_list: for read_method in pandas_black_list:
...@@ -899,4 +932,4 @@ def test_suite(): ...@@ -899,4 +932,4 @@ def test_suite():
import AccessControl.tests.testZopeSecurityPolicy import AccessControl.tests.testZopeSecurityPolicy
suite.addTest(AccessControl.tests.testZopeSecurityPolicy.test_suite()) suite.addTest(AccessControl.tests.testZopeSecurityPolicy.test_suite())
return suite return suite
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment