def test_handle_month_quarter(self): transform_col = self.init_series("2000-05-01", constants.PANDAS_MONTH, constants.PANDAS_QUARTER) prev_length = len(transform_col) handle_missing_values(self.col, transform_col) self.assertEqual(len(transform_col), prev_length - 2)
def test_handle_day_month(self): transform_col = self.init_series("2000-06-05", constants.PANDAS_DAY, constants.PANDAS_MONTH) prev_length = len(transform_col) handle_missing_values(self.col, transform_col) self.assertEqual(len(transform_col), prev_length - 2)
def test_handle_semester_year(self): transform_col = self.init_series("2000-02-01", constants.PANDAS_SEMESTER, constants.PANDAS_YEAR) prev_length = len(transform_col) handle_missing_values(self.col, transform_col) self.assertEqual(len(transform_col), prev_length - 2)