def test_reify(self, mini_amr):
m = Model()
with pytest.raises(ModelError):
m.reify(('a', ':ARG0', 'b'))
with pytest.raises(ModelError):
m.reify(('a', ':accompanier', 'b'))
with pytest.raises(ModelError):
m.reify(('a', ':domain', 'b'))
with pytest.raises(ModelError):
m.reify(('a', ':mod', 'b'))
m = Model.from_dict(mini_amr)
with pytest.raises(ModelError):
m.reify(('a', ':ARG0', 'b'))
assert m.reify(
('a', ':accompanier',
'b')) == (('_', ':ARG0', 'a'), ('_', ':instance', 'accompany-01'),
('_', ':ARG1', 'b'))
with pytest.raises(ModelError):
assert m.reify(('a', ':domain', 'b'))
assert m.reify(
('a', ':mod', 'b')) == (('_', ':ARG1', 'a'), ('_', ':instance',
'have-mod-91'),
('_', ':ARG2', 'b'))
# ensure unique ids if variables is specified
assert m.reify(('a', ':mod', 'b'),
variables={'a', 'b',
'_'}) == (('_2', ':ARG1', 'a'),
('_2', ':instance', 'have-mod-91'),
('_2', ':ARG2', 'b'))
def test_canonicalize_role(self, mini_amr):
m = Model()
assert m.canonicalize_role(':ARG0') == ':ARG0'
assert m.canonicalize_role(':ARG0-of') == ':ARG0-of'
assert m.canonicalize_role(':ARG0-of-of') == ':ARG0'
assert m.canonicalize_role(':consist') == ':consist'
assert m.canonicalize_role(':consist-of') == ':consist-of'
assert m.canonicalize_role(':consist-of-of') == ':consist'
assert m.canonicalize_role(':mod') == ':mod'
assert m.canonicalize_role(':mod-of') == ':mod-of'
assert m.canonicalize_role(':domain') == ':domain'
assert m.canonicalize_role(':domain-of') == ':domain-of'
# without :
assert m.canonicalize_role('ARG0') == ':ARG0'
assert m.canonicalize_role('ARG0-of') == ':ARG0-of'
assert m.canonicalize_role('ARG0-of-of') == ':ARG0'
m = Model.from_dict(mini_amr)
assert m.canonicalize_role(':ARG0') == ':ARG0'
assert m.canonicalize_role(':ARG0-of') == ':ARG0-of'
assert m.canonicalize_role(':ARG0-of-of') == ':ARG0'
assert m.canonicalize_role(':consist') == ':consist-of-of'
assert m.canonicalize_role(':consist-of') == ':consist-of'
assert m.canonicalize_role(':consist-of-of') == ':consist-of-of'
assert m.canonicalize_role(':mod') == ':mod'
assert m.canonicalize_role(':mod-of') == ':domain'
assert m.canonicalize_role(':domain') == ':domain'
assert m.canonicalize_role(':domain-of') == ':mod'
# without :
assert m.canonicalize_role('consist') == ':consist-of-of'
assert m.canonicalize_role('consist-of') == ':consist-of'
assert m.canonicalize_role('consist-of-of') == ':consist-of-of'
def test_canonicalize(self, mini_amr):
m = Model()
assert m.canonicalize(('a', ':ARG0', 'b')) == ('a', ':ARG0', 'b')
assert m.canonicalize(('a', ':ARG0-of', 'b')) == ('a', ':ARG0-of', 'b')
assert m.canonicalize(('a', ':ARG0-of-of', 'b')) == ('a', ':ARG0', 'b')
assert m.canonicalize(('a', ':consist', 'b')) == ('a', ':consist', 'b')
assert m.canonicalize(('a', ':consist-of', 'b')) == ('a', ':consist-of', 'b')
assert m.canonicalize(('a', ':consist-of-of', 'b')) == ('a', ':consist', 'b')
assert m.canonicalize(('a', ':mod', 'b')) == ('a', ':mod', 'b')
assert m.canonicalize(('a', ':mod-of', 'b')) == ('a', ':mod-of', 'b')
assert m.canonicalize(('a', ':domain', 'b')) == ('a', ':domain', 'b')
assert m.canonicalize(('a', ':domain-of', 'b')) == ('a', ':domain-of', 'b')
# without :
assert m.canonicalize(('a', 'ARG0', 'b')) == ('a', ':ARG0', 'b')
assert m.canonicalize(('a', 'ARG0-of', 'b')) == ('a', ':ARG0-of', 'b')
assert m.canonicalize(('a', 'ARG0-of-of', 'b')) == ('a', ':ARG0', 'b')
m = Model.from_dict(mini_amr)
assert m.canonicalize(('a', ':ARG0', 'b')) == ('a', ':ARG0', 'b')
assert m.canonicalize(('a', ':ARG0-of', 'b')) == ('a', ':ARG0-of', 'b')
assert m.canonicalize(('a', ':ARG0-of-of', 'b')) == ('a', ':ARG0', 'b')
assert m.canonicalize(('a', ':consist', 'b')) == ('a', ':consist-of-of', 'b')
assert m.canonicalize(('a', ':consist-of', 'b')) == ('a', ':consist-of', 'b')
assert m.canonicalize(('a', ':consist-of-of', 'b')) == ('a', ':consist-of-of', 'b')
assert m.canonicalize(('a', ':mod', 'b')) == ('a', ':mod', 'b')
assert m.canonicalize(('a', ':mod-of', 'b')) == ('a', ':domain', 'b')
assert m.canonicalize(('a', ':domain', 'b')) == ('a', ':domain', 'b')
assert m.canonicalize(('a', ':domain-of', 'b')) == ('a', ':mod', 'b')
# without :
assert m.canonicalize(('a', 'consist', 'b')) == ('a', ':consist-of-of', 'b')
assert m.canonicalize(('a', 'consist-of', 'b')) == ('a', ':consist-of', 'b')
assert m.canonicalize(('a', 'consist-of-of', 'b')) == ('a', ':consist-of-of', 'b')
def test_is_concept_dereifiable(self, mini_amr):
m = Model()
assert not m.is_concept_dereifiable('chase-01')
assert not m.is_concept_dereifiable(':mod')
assert not m.is_concept_dereifiable('have-mod-91')
m = Model.from_dict(mini_amr)
assert not m.is_concept_dereifiable('chase-01')
assert not m.is_concept_dereifiable(':mod')
assert m.is_concept_dereifiable('have-mod-91')
def test_is_role_reifiable(self, mini_amr):
m = Model()
assert not m.is_role_reifiable(':ARG0')
assert not m.is_role_reifiable(':accompanier')
assert not m.is_role_reifiable(':domain')
assert not m.is_role_reifiable(':mod')
m = Model.from_dict(mini_amr)
assert not m.is_role_reifiable(':ARG0')
assert m.is_role_reifiable(':accompanier')
assert not m.is_role_reifiable(':domain')
assert m.is_role_reifiable(':mod')
def test_invert_role(self, mini_amr):
m = Model()
assert m.invert_role(':ARG0') == ':ARG0-of'
assert m.invert_role(':ARG0-of') == ':ARG0'
assert m.invert_role(':consist-of') == ':consist'
assert m.invert_role(':mod') == ':mod-of'
assert m.invert_role(':domain') == ':domain-of'
# # without :
# assert m.invert_role('ARG0') == 'ARG0-of'
# assert m.invert_role('ARG0-of') == 'ARG0'
m = Model.from_dict(mini_amr)
assert m.invert_role(':ARG0') == ':ARG0-of'
assert m.invert_role(':ARG0-of') == ':ARG0'
assert m.invert_role(':consist-of') == ':consist-of-of'
assert m.invert_role(':mod') == ':mod-of'
assert m.invert_role(':domain') == ':domain-of'
def test_errors(self, mini_amr):
m = Model()
a = Model.from_dict(mini_amr)
# basic roles
g = Graph([('a', ':instance', 'alpha')])
assert m.errors(g) == {}
g = Graph([('a', ':instance', 'alpha'), ('a', ':mod', '1')])
assert m.errors(g) == {('a', ':mod', '1'): ['invalid role']}
assert a.errors(g) == {}
# regex role names
g = Graph([('n', ':instance', 'name'), ('n', ':op1', 'Foo'),
('n', ':op2', 'Bar')])
assert a.errors(g) == {}
# disconnected graph
g = Graph([('a', ':instance', 'alpha'), ('b', ':instance', 'beta')])
assert m.errors(g) == {('b', ':instance', 'beta'): ['unreachable']}
assert a.errors(g) == {('b', ':instance', 'beta'): ['unreachable']}
def test_invert(self, mini_amr):
m = Model()
assert m.invert(('a', ':ARG0', 'b')) == ('b', ':ARG0-of', 'a')
assert m.invert(('a', ':ARG0-of', 'b')) == ('b', ':ARG0', 'a')
assert m.invert(('a', ':consist-of', 'b')) == ('b', ':consist', 'a')
assert m.invert(('a', ':mod', 'b')) == ('b', ':mod-of', 'a')
assert m.invert(('a', ':domain', 'b')) == ('b', ':domain-of', 'a')
# # without :
# assert m.invert(('a', 'ARG0', 'b')) == ('b', 'ARG0-of', 'a')
# assert m.invert(('a', 'ARG0-of', 'b')) == ('b', 'ARG0', 'a')
m = Model.from_dict(mini_amr)
assert m.invert(('a', ':ARG0', 'b')) == ('b', ':ARG0-of', 'a')
assert m.invert(('a', ':ARG0-of', 'b')) == ('b', ':ARG0', 'a')
assert m.invert(('a', ':consist-of', 'b')) == ('b', ':consist-of-of', 'a')
assert m.invert(('a', ':mod', 'b')) == ('b', ':mod-of', 'a')
assert m.invert(('a', ':domain', 'b')) == ('b', ':domain-of', 'a')
def test_is_role_inverted(self, mini_amr):
m = Model()
assert m.is_role_inverted(':ARG0-of')
assert m.is_role_inverted(':-of')
assert not m.is_role_inverted(':ARG0')
assert not m.is_role_inverted(':')
assert m.is_role_inverted(':consist-of')
# # without :
# assert m.is_role_inverted('ARG0-of')
# assert not m.is_role_inverted('ARG0')
m = Model.from_dict(mini_amr)
assert m.is_role_inverted(':mod-of')
assert m.is_role_inverted(':domain-of')
assert not m.is_role_inverted(':mod')
assert not m.is_role_inverted(':domain')
assert m.is_role_inverted(':consist-of-of')
assert not m.is_role_inverted(':consist-of')
def test_dereify(self, mini_amr):
# (a :ARG1-of (_ / age-01 :ARG2 b)) -> (a :age b)
t1 = ('_', ':instance', 'have-mod-91')
t1b = ('_', ':instance', 'chase-01')
t2 = ('_', ':ARG1', 'a')
t3 = ('_', ':ARG2', 'b')
m = Model()
with pytest.raises(TypeError):
m.dereify(t1)
with pytest.raises(TypeError):
m.dereify(t1, t2)
with pytest.raises(ModelError):
m.dereify(t1, t2, t3)
m = Model.from_dict(mini_amr)
assert m.dereify(t1, t2, t3) == ('a', ':mod', 'b')
assert m.dereify(t1, t3, t2) == ('a', ':mod', 'b')
with pytest.raises(ModelError):
m.dereify(t1b, t2, t3)
def test_has_role(self, mini_amr):
m = Model()
assert not m.has_role('')
assert m.has_role(m.concept_role)
assert not m.has_role(':ARG0')
assert not m.has_role(':ARG0-of')
m = Model.from_dict(mini_amr)
assert not m.has_role('')
assert m.has_role(m.concept_role)
assert m.has_role(':ARG0')
assert m.has_role(':ARG0-of')
assert m.has_role(':mod')
assert m.has_role(':mod-of')
assert not m.has_role(':consist')
assert m.has_role(':consist-of')
assert m.has_role(':consist-of-of')
assert not m.has_role(':fake')
assert m.has_role(':op1')
assert m.has_role(':op10')
assert m.has_role(':op9999')
assert not m.has_role(':op[0-9]+')
def test_from_dict(self, mini_amr):
assert Model.from_dict(mini_amr) == Model(
roles=mini_amr['roles'],
normalizations=mini_amr['normalizations'],
reifications=mini_amr['reifications'])
def amr_model(mini_amr):
return Model.from_dict(mini_amr)