Skip to content

Commit bb79f6e

Browse files
committed
fix formatting.
1 parent 41aeaa2 commit bb79f6e

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/test_packets.py

+4-6
Original file line numberDiff line numberDiff line change
@@ -590,17 +590,15 @@ def test_separable_conv_packets_2d(axes: tuple[int, int]) -> None:
590590
assert np.allclose(signal, ptwp[""].data[:, :32, :32, :32])
591591

592592

593-
594593
def test_partial_reconstruction() -> None:
595-
594+
"""Reconstruct a cosine wave from packet filters."""
596595
signal = np.random.randn(1, 16)
597596
signal2 = np.cos(np.linspace(0, 2 * np.pi, 16))
598-
ptwp = WaveletPacket(torch.from_numpy(signal), "haar",
599-
mode="reflect", maxlevel=2)
597+
ptwp = WaveletPacket(torch.from_numpy(signal), "haar", mode="reflect", maxlevel=2)
600598
ptwp.initialize(["aa", "ad", "da", "dd"])
601-
599+
602600
ptwp2 = WaveletPacket(torch.from_numpy(signal2), "haar", mode="reflect", maxlevel=2)
603-
601+
604602
# overwrite the first packet set.
605603
ptwp["aa"] = ptwp2["aa"]
606604
ptwp["ad"] = ptwp2["ad"]

0 commit comments

Comments
 (0)