def test_send(self):
dgraphapi.drop_all()
data = dataset.Dataset()
batch_consumer = consumer.BatchConsumer(data)
data.triple(self.sub, self.pred, self.obj)
data.triple(self.sub, self.pred,
rdfuri.Literal("literal data2", "xs:string"))
result = batch_consumer.send()
self.assertFalse(data)
def test_batch(self):
dgraphapi.drop_all()
data = dataset.Dataset()
processor = batcher.BatchProcessor(data, batch_size=10)
batch_count = 0
for i in range(1, 100):
data.triple(self.sub, self.pred,
rdfuri.Literal("literal data-%s" % i, "xs:string"))
processor.increment()
if processor.count == 0:
batch_count += 1
self.assertTrue(batch_count > 4)
def tearDownClass(cls):
dgraphapi.drop_all()
def setUpClass(cls):
dgraphapi.drop_all()
dgraphapi.add_xid_to_schema()
def test_find_uid_for_xid_with_no_xid_schema(self):
dgraphapi.drop_all()
uid = dgraphapi.create_uid_for_xid("http://test.com")
dgraphapi.drop_all()
dgraphapi.add_xid_to_schema()