def test_tree_properties(sample):
c = 0.08
tree = BIC(c, keep_data=True).fit(sample).context_tree
assert tree.to_str() == '000 1 10 100'
integrity.is_freq_consistent(tree)
integrity.satisfies_irreductibility(tree)
integrity.satisfies_completeness(tree, sample)
integrity.check_admissibility(tree, sample)
def run_ctm(X, args):
logging.info("Estimating BIC tree:")
# Instantiates BIC object with received parameters
bic = BIC(args.penalty,
df_method=args.df,
keep_data=keep(args),
perl_compatible=args.perl_compatible)
bic.fit(X)
tree = bic.context_tree
logging.info("Tree found:")
logging.info(tree.to_str(reverse=True))
save_file(tree, args)
check_consistency(tree, args)
def test_sample_ep_empty_tree(sample_ep):
bic = BIC(200).fit(sample_ep)
assert bic.context_tree.to_str() == ''
def test_lipsum(sample_lipsum):
bic = BIC(0).fit(sample_lipsum)
s = bic.context_tree.generate_sample(400)
print(s)
def test_tree_estimation(sample):
t1 = '000000 000010 000100 001000 001010 1 10000 100000 100010 10010 100100 10100 101000 101010'
t2 = ''
assert BIC(0.0, keep_data=True).fit(sample).context_tree.to_str() == t1
assert BIC(2000, keep_data=True).fit(sample).context_tree.to_str() == t2
def test_sample_ep(sample_ep):
bic = BIC(0.2654890885033804).fit(sample_ep)
t = '0000 2000 100 200 300 0010 2010 210 20 30 001 201 21 2 3 4'
assert bic.context_tree.to_str(reverse=True) == t
def test_sample_bp(sample_bp):
bic = BIC(0.2982733078507897).fit(sample_bp)
t = '0000 2000 100 200 300 0010 2010 210 20 30 001 201 21 2 3 4'
assert bic.context_tree.to_str(reverse=True) == t
def test_perl_compatibility(sample_pl_compat):
bic = BIC(2000, keep_data=False, perl_compatible=True)
bic.fit(sample_pl_compat)
# perl version doesn't have the empty node
assert bic.context_tree.to_str() == '0 1'
def test_sample_generation(sample):
tree = BIC(0.08).fit(sample).context_tree
s = tree.generate_sample(1000)
assert len(s) == 1000