def test_errors(self, value):
message = "this is a message"
with pytest.raises(TypeError):
species_trees.parse_number_or_mapping(value, message)
def test_simple_mapping(self):
arg = {"x": 1, "y": 2}
res = species_trees.parse_number_or_mapping(arg, "")
assert arg is res
def test_defaultdict(self):
arg = collections.defaultdict(lambda x: 1234)
res = species_trees.parse_number_or_mapping(arg, "")
assert arg is res
def test_single_value(self, N):
res = species_trees.parse_number_or_mapping(N, "")
# This is a defaultdict, so we should get the right value regardless
# of the input.
for key in ["x", "", None, 123]:
assert res[key] == N