improve coverage

diffsptk/misc/utils.py

@@ -53,27 +53,27 @@ def next_power_of_two(n):
 
 def default_dtype():
     t = torch.get_default_dtype()
-    if t == torch.float:  # pragma: no cover
+    if t == torch.float:
         return np.float32
-    elif t == torch.double:  # pragma: no cover
+    elif t == torch.double:
         return np.float64
     raise RuntimeError("Unknown default dtype: {t}.")
 
 
 def default_complex_dtype():
     t = torch.get_default_dtype()
-    if t == torch.float:  # pragma: no cover
+    if t == torch.float:
         return np.complex64
-    elif t == torch.double:  # pragma: no cover
+    elif t == torch.double:
         return np.complex128
     raise RuntimeError("Unknown default dtype: {t}.")
 
 
 def torch_default_complex_dtype():
     t = torch.get_default_dtype()
-    if t == torch.float:  # pragma: no cover
+    if t == torch.float:
         return torch.complex64
-    elif t == torch.double:  # pragma: no cover
+    elif t == torch.double:
         return torch.complex128
     raise RuntimeError("Unknown default dtype: {t}.")
 

pyproject.toml

@@ -88,6 +88,8 @@ exclude_lines = [
   "verbose",
   "warn_type",
   "warnings",
+  "torch.float",
+  "torch.double",
 ]
 
 [tool.coverage.run]

tests/test_igammatone.py

@@ -43,7 +43,7 @@ def test_analysis_synthesis(device, exact, M=4, L=8192, desired_delay=4, verbose
     igammatone = diffsptk.GammatoneFilterBankSynthesis(
         sr, filter_order=M, exact=exact, desired_delay=desired_delay
     ).to(device)
-    y = igammatone(gammatone(x.unsqueeze(0)), compensate_delay=True)
+    y = igammatone(gammatone(x.unsqueeze(0)).squeeze(0), compensate_delay=True)
     assert (x - y).abs().max() < 0.1
 
     if verbose: