def test_spaces(unittest):
df = replacements_data()
data_id, replacement_type = "1", "spaces"
with ExitStack() as stack:
stack.enter_context(
mock.patch("dtale.global_state.DATA", {data_id: df}))
builder = ColumnReplacement(data_id, "b", replacement_type, {})
verify_builder(
builder,
lambda col: unittest.assertEqual(list(col.values),
["", np.nan, " - "]),
)
builder = ColumnReplacement(data_id, "b", replacement_type,
{"value": "blah"})
verify_builder(
builder,
lambda col: unittest.assertEqual(list(col.values),
["", "blah", " - "]),
)
def test_simple_imputers(unittest):
df = replacements_data()
data_id, replacement_type = "1", "imputer"
with ExitStack() as stack:
stack.enter_context(
mock.patch("dtale.global_state.DATA", {data_id: df}))
cfg = {"type": "simple"}
builder = ColumnReplacement(data_id, "d", replacement_type, cfg)
verify_builder(
builder,
lambda col: unittest.assertEqual(list(col.values), [1.1, 2.05, 3]))