def testGetCompaniesForIndex(self):
COMPANY_NAMES = ["IBM", "JCB", "ICL", "JPM"]
i = Node("Index", name="Test Index")
I = Index(i)
for company_name in COMPANY_NAMES:
c = Node("Company", name=company_name)
r = Relationship(i, "INCLUDES", c)
self.g.create(r)
self.assertEqual(len(list(I.get_companies())), 4)
def testWrappedNodeMustBeSomething(self):
with self.assertRaises(ValueError):
Index(None)