ia-guiao-rc/tests/test_aula6.py

80 lines
2.9 KiB
Python

import pytest
from semantic_network import *
@pytest.fixture
def sn_net():
a = Association('socrates','professor','filosofia')
s = Subtype('homem','mamifero')
m = Member('socrates','homem')
da = Declaration('descartes',a)
ds = Declaration('darwin',s)
dm = Declaration('descartes',m)
z = SemanticNetwork()
z.insert(da)
z.insert(ds)
z.insert(dm)
z.insert(Declaration('darwin',Association('mamifero','mamar','sim')))
z.insert(Declaration('darwin',Association('homem','gosta','carne')))
# novas declaracoes
z.insert(Declaration('darwin',Subtype('mamifero','vertebrado')))
z.insert(Declaration('descartes', Member('aristoteles','homem')))
b = Association('socrates','professor','matematica')
z.insert(Declaration('descartes',b))
z.insert(Declaration('simao',b))
z.insert(Declaration('simoes',b))
z.insert(Declaration('descartes', Member('platao','homem')))
e = Association('platao','professor','filosofia')
z.insert(Declaration('descartes',e))
z.insert(Declaration('simao',e))
z.insert(Declaration('descartes',Association('mamifero','altura',1.2)))
z.insert(Declaration('descartes',Association('homem','altura',1.75)))
z.insert(Declaration('simao',Association('homem','altura',1.85)))
z.insert(Declaration('darwin',Association('homem','altura',1.75)))
z.insert(Declaration('descartes', Association('socrates','peso',80)))
z.insert(Declaration('darwin', Association('socrates','peso',75)))
z.insert(Declaration('darwin', Association('platao','peso',75)))
z.insert(Declaration('damasio', Association('filosofo','gosta','filosofia')))
z.insert(Declaration('damasio', Member('socrates','filosofo')))
return z
def test_exercicio1(sn_net):
assert sorted(sn_net.list_associations()) == ['altura', 'gosta', 'mamar', 'peso', 'professor']
def test_exercicio2(sn_net):
assert sorted(sn_net.list_objects()) == ['aristoteles', 'platao', 'socrates']
def test_exercicio3(sn_net):
assert sorted(sn_net.list_users()) == ['damasio', 'darwin', 'descartes', 'simao', 'simoes']
def test_exercicio4(sn_net):
assert sorted(sn_net.list_types()) == ['filosofo', 'homem', 'mamifero', 'vertebrado']
def test_exercicio5(sn_net):
assert sorted(sn_net.list_local_associations('socrates')) == ['peso', 'professor']
def test_exercicio6(sn_net):
assert sorted(sn_net.list_relations_by_user('descartes')) == ['altura', 'member', 'peso', 'professor']
def test_exercicio7(sn_net):
assert sn_net.associations_by_user('descartes') == 3
def test_exercicio8(sn_net):
assert sorted(sn_net.list_local_associations_by_entity('socrates')) == [('peso', 'darwin'), ('peso', 'descartes'), ('professor', 'descartes'), ('professor', 'simao'), ('professor', 'simoes')]
def test_exercicio9(sn_net):
assert sn_net.predecessor('vertebrado','socrates')
assert not sn_net.predecessor('vertebrado','filosofo')