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()