1import numpy as np
2from zia_benchmark.algorithms.simple_ans.markov import markov_predict, markov_reconstruct
3from zia_benchmark.algorithms.simple_ans.markov_reconstruct_wrapper import markov_reconstruct as markov_reconstruct_cpp
5coeffs = np.array([1, 2, 3], dtype=np.float32)
6initial = np.array([7, 5], dtype=np.int16)
7resid = np.array([6, 7], dtype=np.int16)
8a = markov_reconstruct(coeffs, initial, resid)
10b = markov_reconstruct_cpp(coeffs, initial, resid)
12print(a)
13print(b)
14print(type(b))
15print(b.shape)
16print(b.dtype)