def test_init_with_dbm():
rdf_space = Space('tests/example.n3', index_dir = 'tests/example-space')
uris = rdf_space._uri_index.keys()
indexes = rdf_space._index_uri.keys()
assert(os.path.isfile('tests/example-space/uri_index.db'))
assert(os.path.isfile('tests/example-space/index_uri.db'))
rdf_space.save()
rdf_space = Space.load('tests/example-space')
assert_equal(rdf_space._uri_index.keys(), uris)
assert_equal(rdf_space._index_uri.keys(), indexes)
def test_save_and_load():
rdf_space = Space('tests/example.n3')
rdf_space._ut = np.random.rand(5, 5)
rdf_space._ut_shape = (5, 5)
rdf_space._s = np.random.rand(5, 5)
rdf_space._s_shape = (5, 5)
rdf_space._vt = np.random.rand(5, 5)
rdf_space._vt_shape = (5, 5)
adj = rdf_space._adjacency
rdf_space.save('tests/example-space')
assert_array_equal(rdf_space._adjacency, adj)
space = Space.load('tests/example-space')
assert_equal(space.uri_index, rdf_space.uri_index)
assert_equal(space._ut[2,3], rdf_space._ut[2,3])
assert_equal(space._s[2,2], rdf_space._s[2,2])
assert_equal(space._vt[2,3], rdf_space._vt[2,3])
assert_equal(space._adjacency, None)
try:
space = Space.load('foo')
assert_true(false)
except Exception as e:
assert_equal(e.args[0], 'No such directory')
