def test_serialize_simple():
N, V = 10, 20
defn = model_definition(N, V)
data = toy_dataset(defn)
view = data
prng = rng()
s = initialize(defn, view, prng)
m = s.serialize()
s2 = deserialize(defn, m)
assert s2.__class__ == s.__class__
def test_serialize_simple():
docs = [list('abcd'), list('cdef')]
defn = model_definition(len(docs), v=6)
prng = rng()
s = initialize(defn, docs, prng)
m = s.serialize()
s2 = deserialize(defn, m)
assert s2.__class__ == s.__class__
assert all(word in "abcdef" for wd in s2.word_distribution_by_topic()
for word in wd.keys())
assert all(
isinstance(word, str) for wd in s2.word_distribution_by_topic()
for word in wd.keys())
def test_serialize_simple():
docs = [list('abcd'), list('cdef')]
defn = model_definition(len(docs), v=6)
prng = rng()
s = initialize(defn, docs, prng)
m = s.serialize()
s2 = deserialize(defn, m)
assert s2.__class__ == s.__class__
assert all(word in "abcdef"
for wd in s2.word_distribution_by_topic()
for word in wd.keys())
assert all(isinstance(word, str)
for wd in s2.word_distribution_by_topic()
for word in wd.keys())
