def test_dither_script(dither_magnitude):
dither = DitherAudio(dither_magnitude)
dither.eval()
x = torch.randn(10, 1337)
dither_script = torch.jit.script(dither)
assert torch.allclose(dither(x), dither_script(x))
def test_dither_trace(dither_magnitude):
dither = DitherAudio(dither_magnitude)
dither.eval()
x = torch.randn(10, 1337)
dither_trace = torch.jit.trace(dither, (x))
assert torch.allclose(dither(x), dither_trace(x))
def test_dither_onnx(dither_magnitude):
dither = DitherAudio(dither_magnitude)
dither.eval()
x = torch.randn(10, 1337)
with TemporaryDirectory() as export_path:
torch.onnx.export(
dither,
(x),
f"{export_path}/Dither.onnx",
verbose=True,
opset_version=11,
)
def test_dither_eval_mode_retain_input(dither_magnitude):
dither = DitherAudio(dither_magnitude)
dither.eval()
x = torch.randn(10, 1337)
out = dither(x)
assert torch.allclose(out, x)
def test_dither_retains_shape(dither_magnitude):
dither = DitherAudio(dither_magnitude)
x = torch.randn(10, 1337)
out = dither(x)
assert out.shape[0] == x.shape[0]
assert out.shape[1] == x.shape[1]
def test_dither_device_move(dither_magnitude):
dither = DitherAudio(dither_magnitude)
x = torch.randn(10, 1337)
_test_device_move(dither, x)
def test_dither_batch_independence(dither_magnitude):
dither = DitherAudio(dither_magnitude)
x = torch.randn(10, 1337)
_test_batch_independence(dither, x)
def test_dither_changes_input(dither_magnitude):
dither = DitherAudio(dither_magnitude)
x = torch.randn(10, 1337)
out = dither(x)
assert not torch.allclose(out, x)