mirror of
https://github.com/neuml/txtai.git
synced 2026-10-12 02:19:52 +00:00
Support graph=True in embeddings config, closes #848
This commit is contained in:
@@ -146,6 +146,16 @@ class TestGraph(unittest.TestCase):
|
||||
# Close graph
|
||||
graph.close()
|
||||
|
||||
def testDefault(self):
|
||||
"""
|
||||
Test embeddings default graph setting
|
||||
"""
|
||||
|
||||
embeddings = Embeddings(content=True, graph=True)
|
||||
embeddings.index([(uid, text, None) for uid, text in enumerate(self.data)])
|
||||
|
||||
self.assertEqual(embeddings.graph.count(), len(self.data))
|
||||
|
||||
def testDelete(self):
|
||||
"""
|
||||
Test delete
|
||||
|
||||
Reference in New Issue
Block a user