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 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 testAddOrNewCompanyToIndex(self):
i = Index.new(self.g, {"name": "Test Index 3"})
self.assertEqual(len(list(i.get_companies())), 0)
i.add_companies([{'name': "CBS"}])
i.add_companies([{'name': "CBS"}], unique=True)
self.assertEqual(len(list(i.get_companies())), 1)
def testAddOrNewCompanyToIndex(self):
i = Index.new(self.g, {"name": "Test Index 3"})
self.assertEqual(len(list(i.get_companies())), 0)
i.add_companies([{'name': "CBS"}])
i.add_companies([{'name': "CBS"}], unique=True)
self.assertEqual(len(list(i.get_companies())), 1)
def testAddUniqueNode(self):
self.assertEqual(len(list(Index.find(self.g, {"name": "Test Index 6"}))), 0)
Index.new(self.g, {"name": "Test Index 6"}, unique=True)
self.assertEqual(len(list(Index.find(self.g, {"name": "Test Index 6"}))), 1)
Index.new(self.g, {"name": "Test Index 6"}, unique=True)
self.assertEqual(len(list(Index.find(self.g, {"name": "Test Index 6"}))), 1)
def testAddUniqueNode(self):
self.assertEqual(
len(list(Index.find(self.g, {"name": "Test Index 6"}))), 0)
Index.new(self.g, {"name": "Test Index 6"}, unique=True)
self.assertEqual(
len(list(Index.find(self.g, {"name": "Test Index 6"}))), 1)
Index.new(self.g, {"name": "Test Index 6"}, unique=True)
self.assertEqual(
len(list(Index.find(self.g, {"name": "Test Index 6"}))), 1)
def testAddedNodesWrapActualNodes(self):
i = Index.new(self.g, {"name": "Test Index"})
c = i.add_companies([{"name": "Fluffer"}])[0]
self.assertIsInstance(c.n, Node)
def testAddCompanyToIndex(self):
i = Index.new(self.g, {"name": "Test Index 6"})
self.assertEqual(len(list(i.get_companies())), 0)
i.add_companies([{'name': "CBS"}, {'name': 'ABC'}])
self.assertEqual(len(list(i.get_companies())), 2)
def testCreateAndFindIndex(self):
index_name = "Test Index 4"
i = Index.new(self.g, {"name": index_name}).push()
ix = Index.find_one(self.g, dict(name=index_name))
self.assertEqual(i, ix)
def testCreateIndexNode(self):
i = Index.new(self.g, {"name": "Test Index 2"})
self.assertIsInstance(i, Index)
self.assertEqual(i.n.properties['name'], "Test Index 2")
def testWrappedNodeMustBeSomething(self):
with self.assertRaises(ValueError):
Index(None)
def testAddedNodesWrapActualNodes(self):
i = Index.new(self.g, {"name": "Test Index"})
c = i.add_companies([{"name": "Fluffer"}])[0]
self.assertIsInstance(c.n, Node)
def testAddCompanyToIndex(self):
i = Index.new(self.g, {"name": "Test Index 6"})
self.assertEqual(len(list(i.get_companies())), 0)
i.add_companies([{'name': "CBS"}, {'name': 'ABC'}])
self.assertEqual(len(list(i.get_companies())), 2)
def testCreateAndFindIndex(self):
index_name = "Test Index 4"
i = Index.new(self.g, {"name": index_name}).push()
ix = Index.find_one(self.g, dict(name=index_name))
self.assertEqual(i, ix)
def testCreateIndexNode(self):
i = Index.new(self.g, {"name": "Test Index 2"})
self.assertIsInstance(i, Index)
self.assertEqual(i.n.properties['name'], "Test Index 2")
