def test_can_override_handlers_for_supertypes():
s = SpecificationMapper()
s.define_specification_for_instances(tuple, lambda s, d: sum(d))
s.define_specification_for_instances(Litter, lambda s, d: len(d))
assert s.specification_for((1, 2)) == 3
assert s.specification_for(Litter(2, 2)) == 2
def test_can_override_handlers_for_supertypes():
s = SpecificationMapper()
s.define_specification_for_instances(tuple, lambda s, d: sum(d))
s.define_specification_for_instances(Litter, lambda s, d: len(d))
assert s.specification_for((1, 2)) == 3
assert s.specification_for(Litter(2, 2)) == 2
def test_can_call_previous_in_overridden_specifications():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda _, s: len(s))
s.define_specification_for_instances(
str, lambda _, s: 5 if len(s) > 5 else next_in_chain())
assert s.specification_for("foo") == 3
assert s.specification_for(
"I am the very model of a modern major general") == 5
def test_defining_new_handlers_resets_cache():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda *_: Foo())
x = s.specification_for("foo")
s.define_specification_for_instances(str, lambda *_: Fooc())
y = s.specification_for("foo")
assert y is not x
assert isinstance(y, Fooc)
def test_defining_new_handlers_resets_cache():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda *_: Foo())
x = s.specification_for("foo")
s.define_specification_for_instances(str, lambda *_: Fooc())
y = s.specification_for("foo")
assert y is not x
assert isinstance(y, Fooc)
def test_cache_correctly_handles_inheritance():
s = SpecificationMapper()
s.define_specification_for_instances(list, lambda s, d: [s.specification_for(d[0])] )
t = s.new_child_mapper()
t.define_specification_for_instances(str, lambda *_: Foo())
x = t.specification_for("foo")
y = t.specification_for(["foo"])[0]
assert x is y
def test_cache_correctly_handles_inheritance():
s = SpecificationMapper()
s.define_specification_for_instances(
list, lambda s, d: [s.specification_for(d[0])])
t = s.new_child_mapper()
t.define_specification_for_instances(str, lambda *_: Foo())
x = t.specification_for("foo")
y = t.specification_for(["foo"])[0]
assert x is y
def test_can_call_previous_in_overridden_specifications():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda _, s: len(s))
s.define_specification_for_instances(
str,
lambda _, s: 5 if len(s) > 5 else next_in_chain()
)
assert s.specification_for("foo") == 3
assert s.specification_for(
"I am the very model of a modern major general"
) == 5
def test_can_handle_large_numbers_of_instance_mappers():
def f(s,x): return str(x)
s = SpecificationMapper()
s.define_specification_for_instances(tuple, f)
s.define_specification_for_instances(Litter, f)
s.define_specification_for_instances(list, f)
s.define_specification_for_instances(set, f)
s.define_specification_for_instances(str, f)
s.define_specification_for_instances(int, f)
assert s.specification_for((1,1)) == "(1, 1)"
def test_can_handle_large_numbers_of_instance_mappers():
def f(s, x):
return str(x)
s = SpecificationMapper()
s.define_specification_for_instances(tuple, f)
s.define_specification_for_instances(Litter, f)
s.define_specification_for_instances(list, f)
s.define_specification_for_instances(set, f)
s.define_specification_for_instances(str, f)
s.define_specification_for_instances(int, f)
assert s.specification_for((1, 1)) == "(1, 1)"
def test_can_define_instance_specifications_for_lists():
sm = SpecificationMapper()
sm.define_specification_for_instances(list, lambda _, l: len(l))
assert sm.specification_for([1, 2]) == 2
def test_can_handle_subtypes_of_instances():
s = SpecificationMapper()
s.define_specification_for_instances(tuple, lambda s, d: sum(d))
assert s.specification_for((1, 2)) == 3
assert s.specification_for(Litter(2, 2)) == 4
def test_multiple_calls_return_same_value():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda *_: Foo())
assert s.specification_for("foo") is s.specification_for("foo")
assert s.specification_for("foo") is not s.specification_for("bar")
def test_multiple_calls_return_same_value():
s = SpecificationMapper()
s.define_specification_for_instances(str, lambda *_: Foo())
assert s.specification_for("foo") is s.specification_for("foo")
assert s.specification_for("foo") is not s.specification_for("bar")
def test_can_override_instance_specifications():
s = SpecificationMapper()
s.define_specification_for_instances(str, const(1))
s.define_specification_for_instances(str, const(2))
assert s.specification_for("foo") == 2
def test_can_define_instance_specifications():
sm = SpecificationMapper()
sm.define_specification_for_instances(str, lambda _, i: i + "bar")
assert sm.specification_for("foo") == "foobar"
def test_can_define_instance_specifications():
sm = SpecificationMapper()
sm.define_specification_for_instances(str, lambda _, i: i + "bar")
assert sm.specification_for("foo") == "foobar"
def test_can_handle_subtypes_of_instances():
s = SpecificationMapper()
s.define_specification_for_instances(tuple, lambda s, d: sum(d))
assert s.specification_for((1, 2)) == 3
assert s.specification_for(Litter(2, 2)) == 4
def test_can_define_instance_specifications_for_lists():
sm = SpecificationMapper()
sm.define_specification_for_instances(list, lambda _, l: len(l))
assert sm.specification_for([1, 2]) == 2
def test_can_override_instance_specifications():
s = SpecificationMapper()
s.define_specification_for_instances(str, const(1))
s.define_specification_for_instances(str, const(2))
assert s.specification_for("foo") == 2
