Skip to content

Commit

Permalink
simplify test
Browse files Browse the repository at this point in the history
  • Loading branch information
kevinsung committed Dec 7, 2023
1 parent f4e99b2 commit 206ab7f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/random_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@ def assert_t2_has_correct_symmetry(t2: np.ndarray):
nocc, _, nvrt, _ = t2.shape
norb = nocc + nvrt
pairs = itertools.product(range(nocc), range(nocc, norb))
for (m, (i, a)), (n, (j, b)) in itertools.product(enumerate(pairs), repeat=2):
if m <= n and i <= j and a <= b:
for (i, a), (j, b) in itertools.product(pairs, repeat=2):
if i <= j and a <= b:
np.testing.assert_allclose(
t2[i, j, a - nocc, b - nocc], t2[j, i, b - nocc, a - nocc]
)
Expand Down

0 comments on commit 206ab7f

Please sign in to comment.