[testing] Fix Infleqtion tests for state prep (#4323)

* Follow-up to PR #3824
* Fix the `assert_close` function.
* Manually tested 
```
python3 -m pytest -v python/tests/backends/test_Infleqtion.py
=================================================================== test session starts ===================================================================
platform linux -- Python 3.12.3, pytest-8.3.0, pluggy-1.6.0 -- /usr/bin/python3
cachedir: .pytest_cache
rootdir: /workspaces/cuda-quantum
configfile: pyproject.toml
plugins: anyio-4.13.0, xdist-3.8.0
collected 9 items                                                                                                                                         

python/tests/backends/test_Infleqtion.py::test_simple_kernel PASSED                                                                                 [ 11%]
python/tests/backends/test_Infleqtion.py::test_all_gates PASSED                                                                                     [ 22%]
python/tests/backends/test_Infleqtion.py::test_multiple_qvector PASSED                                                                              [ 33%]
python/tests/backends/test_Infleqtion.py::test_multiple_measure PASSED                                                                              [ 44%]
python/tests/backends/test_Infleqtion.py::test_observe PASSED                                                                                       [ 55%]
python/tests/backends/test_Infleqtion.py::test_state_synthesis PASSED                                                                               [ 66%]
python/tests/backends/test_Infleqtion.py::test_state_preparation PASSED                                                                             [ 77%]
python/tests/backends/test_Infleqtion.py::test_state_preparation_builder PASSED                                                                     [ 88%]
python/tests/backends/test_Infleqtion.py::test_exp_pauli PASSED                                                                                     [100%]

=================================================================== 9 passed in 18.45s ====================================================================

```

---------

Signed-off-by: Pradnya Khalate <pkhalate@nvidia.com>

Author: khalatepradnya

python/tests/backends/test_Infleqtion.py

@@ -24,8 +24,8 @@ def set_up_target():
     cudaq.reset_target()
 
 
-def assert_close(got) -> bool:
-    return got < -1.5 and got > -1.9
+def assert_close(want, got, tolerance=1.0e-5) -> bool:
+    return abs(want - got) < tolerance
 
 
 def test_simple_kernel():
@@ -179,31 +179,28 @@ def kernel2(s: cudaq.State):
 
 
 def test_state_preparation():
-    shots = 10000
+    shots = 100
 
     @cudaq.kernel
     def kernel(vec: list[complex]):
         qubits = cudaq.qvector(vec)
 
     state = [1. / np.sqrt(2.), 1. / np.sqrt(2.), 0., 0.]
     counts = cudaq.sample(kernel, state, shots_count=shots)
-    assert assert_close(counts["00"], shots / 2, 2)
-    assert assert_close(counts["10"], shots / 2, 2)
-    assert assert_close(counts["01"], 0., 2)
-    assert assert_close(counts["11"], 0., 2)
+    counts.dump()
+    assert assert_close(shots / 2, counts["00"], shots / 10)
+    assert assert_close(shots / 2, counts["10"], shots / 10)
 
 
 def test_state_preparation_builder():
-    shots = 10000
+    shots = 100
     kernel, state = cudaq.make_kernel(list[complex])
     qubits = kernel.qalloc(state)
 
     state = [1. / np.sqrt(2.), 1. / np.sqrt(2.), 0., 0.]
     counts = cudaq.sample(kernel, state, shots_count=shots)
-    assert assert_close(counts["00"], shots / 2, 2)
-    assert assert_close(counts["10"], shots / 2, 2)
-    assert assert_close(counts["01"], 0., 2)
-    assert assert_close(counts["11"], 0., 2)
+    assert assert_close(shots / 2, counts["00"], shots / 10)
+    assert assert_close(shots / 2, counts["10"], shots / 10)
 
 
 def test_exp_pauli():