def test_invalid_expansion(self): v = Variant("variant name") with pytest.raises(TypeError): v.expansion(42, "v") with pytest.raises(TypeError): v.expansion("k", 42)
def test_expansions_can_be_added(self): v = Variant("variant name") v.expansion("k0", "v0") v.expansions({"k1": "v1", "k2": "v2"}) obj = v.to_map() assert "v0" in obj["expansions"]["k0"] assert "v1" in obj["expansions"]["k1"] assert "v2" in obj["expansions"]["k2"]