def scale_scale_discrete_mapping_nulls():
a = np.array([1, 2, 3], dtype=object)
sc = _scale_manual([1, 2, 3, 4, 5])
sc.train(a)
res = sc.map([1, 2, 3])
expected = np.array([1, 2, 3])
npt.assert_array_equal(res, expected)
sc = _scale_manual([1, None, 3, 4, 5])
sc.train(a)
res = sc.map([1, 2, 3])
expected = np.array([1, np.nan, 3])
assert res[0] == expected[0]
assert res[1] is expected[1]
assert res[2] == expected[2]
def scale_scale_discrete_mapping_nulls():
a = np.array([1, 2, 3], dtype=object)
sc = _scale_manual([1, 2, 3, 4, 5])
sc.train(a)
res = sc.map([1, 2, 3])
expected = np.array([1, 2, 3])
npt.assert_array_equal(res, expected)
sc = _scale_manual([1, None, 3, 4, 5])
sc.train(a)
res = sc.map([1, 2, 3])
expected = np.array([1, np.nan, 3])
assert res[0] == expected[0]
assert res[1] is expected[1]
assert res[2] == expected[2]
