diff --git a/unittests/test_transformers.py b/unittests/test_transformers.py index 6b1c4655d7e28a4949d29ce6e344d104c9f74ced..a2d227adc5b0c6a8f2f96cb054e1c7670e980e10 100644 --- a/unittests/test_transformers.py +++ b/unittests/test_transformers.py @@ -168,20 +168,22 @@ def test_cast_transformer_functions(): for val in ("True", "true", "False", "false"): assert type(cast_to_bool(val, {})) == bool if val[1] == "r": - assert cast_to_bool(val, {}) + assert cast_to_bool(val, {}) is True else: - assert not cast_to_bool(val, {}) + assert cast_to_bool(val, {}) is False for val_err in ("jaksdlfj", "0", 1): with pytest.raises(ValueError): cast_to_bool(val_err, {}) - assert not cast_to_bool(False, {}) - assert cast_to_bool(True, {}) + assert cast_to_bool(False, {}) is False + assert cast_to_bool(True, {}) is True assert cast_to_int("24", {}) == 24 assert cast_to_int(24.0, {}) == 24 assert cast_to_int(24, {}) == 24 + assert cast_to_int("-24", {}) == -24 with pytest.raises(ValueError): cast_to_int("24dsf", {}) + with pytest.raises(ValueError): cast_to_int("24.0", {}) == 24 assert cast_to_float("24", {}) == 24.0