From 400686439d3486c2348a5e0b7b521ac87be69bd3 Mon Sep 17 00:00:00 2001
From: cchenhaifeng <1090056852@qq.com>
Date: Tue, 25 Feb 2025 18:13:50 +0800
Subject: [PATCH] fix codestyle

---
 tests/unit/tts/test_losses.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/unit/tts/test_losses.py b/tests/unit/tts/test_losses.py
index f883d5e90..f99d15d1c 100644
--- a/tests/unit/tts/test_losses.py
+++ b/tests/unit/tts/test_losses.py
@@ -30,14 +30,14 @@ def test_multi_scale_stft_loss():
     x, y = get_input()
     loss = MultiScaleSTFTLoss()
     pd_loss = loss(x, y)
-    np.allclose(pd_loss.numpy(), 7.562150, rtol=1e-06)
+    assert np.abs(pd_loss.numpy() - 7.562150) < 1e-06
 
 
 def test_sisdr_loss():
     x, y = get_input()
     loss = SISDRLoss()
     pd_loss = loss(x, y)
-    np.allclose(pd_loss.numpy(), -145.377640, rtol=1e-06)
+    assert np.abs(pd_loss.numpy() - (-145.377640)) < 1e-06
 
 
 def test_gan_loss():
@@ -52,10 +52,10 @@ def test_gan_loss():
     x, y = get_input()
     loss = GANLoss(My_discriminator0())
     pd_loss0, pd_loss1 = loss(x, y)
-    np.allclose(pd_loss0.numpy(), -0.102722, rtol=1e-06)
-    np.allclose(pd_loss1.numpy(), -0.001027, rtol=1e-06)
+    assert np.abs(pd_loss0.numpy() - (-0.102722)) < 1e-06
+    assert np.abs(pd_loss1.numpy() - (-0.001027)) < 1e-06
     loss = GANLoss(My_discriminator1())
     pd_loss0, _ = loss.generator_loss(x, y)
-    np.allclose(pd_loss0.numpy(), 1.000199, rtol=1e-06)
+    assert np.abs(pd_loss0.numpy() - 1.000199) < 1e-06
     pd_loss = loss.discriminator_loss(x, y)
-    np.allclose(pd_loss.numpy(), 1.000200, rtol=1e-06)
+    assert np.abs(pd_loss.numpy() - 1.000200) < 1e-06