def test_linear_mapping_from_string():
m = LinearMapping.from_string("map( -1:3,2.5: 4)")
assert m.input_domain == (-1.0, 3.0)
assert m.output_domain == (2.5, 4.0)
with pytest.raises(MappingError):
LinearMapping.from_string("braboz")
with pytest.raises(MappingError):
LinearMapping.from_string("map(a:b,3:4)")
def test_good_mappings():
sets = [
((1, 5), (1, 5)),
((1, 5), (5, 1)),
((1, 5), (2, 6)),
((1, 5), (0, 10))
]
for inrange, outrange in sets:
m = LinearMapping(inrange, outrange)
assert m.transform(inrange[0]) == outrange[0]
assert m.transform(inrange[1]) == outrange[1]
def test_linear_mapping_strings():
m = LinearMapping((1, 5), (2, 6))
assert m.transform('1') == 2
