volksdata_lua/test.lua

term = require "volksdata.term"
triple = require "volksdata.triple"
graph = require "volksdata.graph"
store = require "volksdata.store"
namespace = require "volksdata.namespace"

---[[
t1 = term.new_bnode()
t2 = term.new_iriref("urn:p:11")
t3 = term.new_lit("123", "xsd:int")
t4 = term.new_lit("Hola", nil, "es_ES")
t5 = term.new_lit("مرحبا", nil, "ar_AR")

trp1 = triple.new (
    term.new_iriref("urn:s:1"),
    term.new_iriref("urn:p:1"),
    term.new_lit("hello", nil, "en_US"))

triples = {
    trp1,
    triple.new (t1, t2, t3),
    triple.new (t1, t2, t4),
    triple.new (t1, t2, t5),
}

mdb_store = store.new(store.MDB, "file:///tmp/volksdata_lua.db", true)
print(("MDB store: %s"):format(mdb_store))
local gr1 = graph.new(mdb_store)

ct = gr1:add(triples)
print("Triples added: " .. ct)
---[[
local gr2 = graph.new(mdb_store)  -- LMDB-backed
local gr3 = graph.new()  -- hashmap-backed
gr1:copy(gr2)
gr1:copy(gr3)
assert(gr1 == gr2)
assert(gr1 == gr3)

--]]
print("All triples:")
for i in gr1:lookup(t1) do print(i) end

---[[
print("Terms connected to t1 t2:")
tset = gr1:term_set(t2, triple.POS_P, t1, triple.POS_S)
for t in pairs(tset) do print(t) end
--]]

---[[
lm = gr1:connections(t1, term.LINK_OUTBOUND)
print("Connections")
for t1, tsiter in pairs(lm) do
    for t2 in tsiter do print(t1, t2) end
end
--]]

---[[
unique_t = gr1:unique_terms(triple.POS_S)
print("Unique subjects")
for t in pairs(unique_t) do print(t) end
print("Unique predicates")
unique_t = gr1:unique_terms(triple.POS_P)
for t in pairs(unique_t) do print(t) end
print("Unique objects")
unique_t = gr1:unique_terms(triple.POS_O)
for t in pairs(unique_t) do print(t) end
--]]

namespace.add("ns1", "urn:ns:1#")
namespace.add("ns2", "urn:ns:2#")
namespace.add("ns3", "urn:ns:3#")

local nsm_map = {}
for k, v in namespace:iter() do nsm_map[k] = v end
assert(nsm_map.ns1 == "urn:ns:1#")
assert(nsm_map.ns2 == "urn:ns:2#")
assert(nsm_map.ns3 == "urn:ns:3#")

local mdb_store_ns = store.new(store.MDB, "file:///tmp/volksdata_nsm.db", true)

mdb_gr2 = graph.new(mdb_store_ns, "ns1:gr2", nsm)
local mdb_gr3 = graph.new(mdb_store_ns, "ns1:gr3", nsm)
local mdb_gr4 = graph.new(mdb_store_ns, "ns1:gr4", nsm)
local mdb_gr5 = graph.new(mdb_store_ns, "ns1:gr5", nsm)
mdb_gr2:add(triples)
mdb_gr3:add(triples)
mdb_gr4:add(triples)

local idx = graph.list(mdb_store_ns)
assert(#idx == 3)

attr = mdb_gr2:attr(t1, t2)
assert (#attr == 3)
assert(attr:contains(term.new_lit("Hola", nil, "es_ES")))
print("Attributes:")
for a in attr:iter() do print(a) end

local gr1_copy = graph.new(mdb_store)
local it = gr1_copy:add_init()
for _, trp in ipairs(triples) do it:add_iter(trp) end
it:add_done()
assert(gr1 == gr1_copy)