diff --git a/.github/trigger_files/beam_PostCommit_XVR_Direct.json b/.github/trigger_files/beam_PostCommit_XVR_Direct.json index 702328d16d4b..2d8ad3760b4b 100644 --- a/.github/trigger_files/beam_PostCommit_XVR_Direct.json +++ b/.github/trigger_files/beam_PostCommit_XVR_Direct.json @@ -1,3 +1,3 @@ { - "modification": 1 + "modification": 2 } diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py index 2e560c013417..8d4bb91036a1 100644 --- a/sdks/python/apache_beam/dataframe/frames_test.py +++ b/sdks/python/apache_beam/dataframe/frames_test.py @@ -2957,17 +2957,21 @@ def test_allow_non_parallel_nesting(self): class ConstructionTimeTest(unittest.TestCase): """Tests for operations that can be executed eagerly.""" - DF = pd.DataFrame({ - 'str_col': ['foo', 'bar'] * 3, - 'int_col': [1, 2] * 3, - 'flt_col': [1.1, 2.2] * 3, - 'cat_col': pd.Series(list('aabbca'), dtype="category"), - 'datetime_col': pd.Series( - pd.date_range( - '1/1/2000', periods=6, freq='m', tz='America/Los_Angeles')) - }) - DEFERRED_DF = frame_base.DeferredFrame.wrap( - expressions.PlaceholderExpression(DF.iloc[:0])) + _DF_COLUMNS = ('str_col', 'int_col', 'flt_col', 'cat_col', 'datetime_col') + + @classmethod + def setUpClass(cls): + cls.DF = pd.DataFrame({ + 'str_col': ['foo', 'bar'] * 3, + 'int_col': [1, 2] * 3, + 'flt_col': [1.1, 2.2] * 3, + 'cat_col': pd.Series(list('aabbca'), dtype="category"), + 'datetime_col': pd.Series( + pd.date_range( + '1/1/2000', periods=6, freq='m', tz='America/Los_Angeles')) + }) + cls.DEFERRED_DF = frame_base.DeferredFrame.wrap( + expressions.PlaceholderExpression(cls.DF.iloc[:0])) def _run_test(self, fn): expected = fn(self.DF) @@ -2982,11 +2986,11 @@ def _run_test(self, fn): else: self.assertEqual(expected, actual) - @parameterized.expand(DF.columns) + @parameterized.expand(_DF_COLUMNS) def test_series_name(self, col_name): self._run_test(lambda df: df[col_name].name) - @parameterized.expand(DF.columns) + @parameterized.expand(_DF_COLUMNS) def test_series_dtype(self, col_name): self._run_test(lambda df: df[col_name].dtype) self._run_test(lambda df: df[col_name].dtypes)