added flatten of state in pytest instead of using within eval_qu

This commit is contained in:
vinitha-balachandran
2024-02-08 16:22:38 +08:00
committed by yangliwei
parent 5131215807
commit 3b73a22e6a

View File

@@ -52,7 +52,7 @@ def test_eval(nqubits: int, tolerance: float, is_mps: bool):
# Test quimb # Test quimb
result_tn = qibotn.eval_qu.dense_vector_tn_qu( result_tn = qibotn.eval_qu.dense_vector_tn_qu(
qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend
) ).flatten()
assert np.allclose( assert np.allclose(
result_sv, result_tn, atol=tolerance result_sv, result_tn, atol=tolerance