From 1ee8ddbc625eccd5b2c5ed954c5a0460519f8ed0 Mon Sep 17 00:00:00 2001 From: Caleb Johnson Date: Fri, 31 Jan 2025 16:23:42 -0600 Subject: [PATCH] Add regression test for recover_configurations (#133) --- test/test_configuration_recovery.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/test/test_configuration_recovery.py b/test/test_configuration_recovery.py index cb43e6a..7e64f39 100644 --- a/test/test_configuration_recovery.py +++ b/test/test_configuration_recovery.py @@ -93,6 +93,19 @@ def test_recover_configurations(self): ) self.assertTrue((expected_mat == mat_rec).all()) self.assertTrue((expected_probs == probs_rec).all()) + with self.subTest("Basic test. Mismatching orbitals."): + bs_mat = np.array([[True, True, True, True]]) + probs = np.array([1.0]) + occs = [0.0, 1.0, 0.0, 0.0] + num_a = 0 + num_b = 1 + expected_mat = np.array([[False, True, False, False]]) + expected_probs = np.array([1.0]) + mat_rec, probs_rec = recover_configurations( + bs_mat, probs, occs, num_a, num_b, rand_seed=4224 + ) + self.assertTrue((expected_mat == mat_rec).all()) + self.assertTrue((expected_probs == probs_rec).all()) with self.subTest("Test with more than 72 bits. Ones to zeros."): n_bits = 74 rng = np.random.default_rng(554)