BUG: F.window_safe implies f.demean().window_safe.

This commit is contained in:
Scott Sanderson
2016-09-22 11:06:35 -04:00
parent a5c7e93540
commit bddb453272
2 changed files with 12 additions and 0 deletions
+11
View File
@@ -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)
+1
View File
@@ -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,
)