mirror of
https://github.com/wassname/catalyst.git
synced 2026-06-29 17:35:15 +08:00
BUG: F.window_safe implies f.demean().window_safe.
This commit is contained in:
@@ -1047,3 +1047,14 @@ class ShortReprTestCase(TestCase):
|
||||
def test_zscore(self):
|
||||
r = F().zscore().short_repr()
|
||||
self.assertEqual(r, "GroupedRowTransform('zscore')")
|
||||
|
||||
|
||||
class TestWindowSafety(TestCase):
|
||||
|
||||
def test_zscore_is_window_safe(self):
|
||||
self.assertTrue(F().zscore().window_safe)
|
||||
|
||||
def test_demean_is_window_safe_if_input_is_window_safe(self):
|
||||
self.assertFalse(F().demean().window_safe)
|
||||
self.assertFalse(F(window_safe=False).demean().window_safe)
|
||||
self.assertTrue(F(window_safe=True).demean().window_safe)
|
||||
|
||||
@@ -509,6 +509,7 @@ class Factor(RestrictedDTypeMixin, ComputableTerm):
|
||||
groupby=groupby,
|
||||
dtype=self.dtype,
|
||||
missing_value=self.missing_value,
|
||||
window_safe=self.window_safe,
|
||||
mask=mask,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user