def test_clean_field_column(self):
eliminated = parser.eliminate_newline_code(self.data)
eliminated = parser.eliminate_rows_with_na_in_economic_status(
eliminated, self.watcher_type.iloc_economic_status_score)
data_with_field = parser.make_field_column(
eliminated, self.watcher_type.iloc_field)
cleaned = parser.clean_field_column(data_with_field)
def test_make_field_column(self):
eliminated = parser.eliminate_newline_code(self.data)
eliminated = parser.eliminate_rows_with_na_in_economic_status(
eliminated, self.watcher_type.iloc_economic_status_score)
data_with_field = parser.make_field_column(
eliminated, self.watcher_type.iloc_field)
self.assertFalse(data_with_field.field.isnull().any())
def __parse_data(data_to_parse: pd.DataFrame, watcher_type: WatcherType):
data = parser.eliminate_rows_with_na_in_economic_status(data_to_parse, watcher_type.iloc_economic_status_score)
data = parser.eliminate_newline_code(data)
data = parser.build_is_tokyo_flag(data, watcher_type.iloc_is_tokyo_flag)
data = parser.make_field_column(data, watcher_type.iloc_field)
data = parser.make_region_column(data)
data = parser.clean_field_column(data)
data = parser.convert_economic_state_score_into_integer(data, watcher_type.iloc_economic_status_score, watcher_type.score_map)
data = parser.eliminate_rows_without_sentence(data, watcher_type.iloc_reason_sentence)
data = parser.clean_sentence_reason(data, watcher_type.iloc_reason_sentence)
return data
def test_make_region_column(self):
eliminated = parser.eliminate_rows_with_na_in_economic_status(
self.data, self.watcher_type.iloc_economic_status_score)
data_with_field = parser.make_field_column(
eliminated, self.watcher_type.iloc_field)
data_with_region = parser.make_region_column(data_with_field)