Remove if __name__ == __main__ from test code

test/functional_cpu_test.py

@@ -405,7 +405,3 @@ def test_mask_along_axis_iid(mask_param, mask_value, axis):
 
     assert mask_specgrams.size() == specgrams.size()
     assert (num_masked_columns < mask_param).sum() == num_masked_columns.numel()
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_batch_consistency.py

@@ -288,7 +288,3 @@ def test_batch_Vol(self):
         # Batch then transform
         computed = torchaudio.transforms.Vol(gain=1.1)(waveform.repeat(3, 1, 1))
         self.assertEqual(computed, expected)
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_compliance_kaldi.py

@@ -246,7 +246,3 @@ def test_resample_waveform_multi_channel(self):
             single_channel_sampled = kaldi.resample_waveform(single_channel, self.test1_signal_sr,
                                                              self.test1_signal_sr // 2)
             torch.testing.assert_allclose(multi_sound_sampled[i, :], single_channel_sampled[0], rtol=1e-4, atol=1e-7)
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_dataloader.py

@@ -37,7 +37,3 @@ def test_1(self):
         dl = DataLoader(ds, batch_size=2)
         for x in dl:
             self.assertTrue(x.size() == expected_size)
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_datasets.py

@@ -171,7 +171,3 @@ def test_libritts(self):
             assert original_text == self.original_text
             assert normalized_text == self.normalized_text
             assert utterance_id == f'{"_".join(str(u) for u in expected_ids[-4:])}'
-
-
-if __name__ == "__main__":
-    unittest.main()

test/test_io.py

@@ -282,6 +282,3 @@ def _test_5_get_info(self):
         self.assertEqual(si.length, samples)
         self.assertEqual(si.rate, rate)
         self.assertEqual(ei.bits_per_sample, precision)
-
-if __name__ == '__main__':
-    unittest.main()

test/test_kaldi_io.py

@@ -33,7 +33,3 @@ def test_read_vec_flt_ark(self):
 
     def test_read_mat_ark(self):
         self._test_helper("mat.ark", [self.data1, self.data2], kio.read_mat_ark, torch.float32)
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_librosa_compatibility.py

@@ -347,7 +347,3 @@ def test_InverseMelScale(self):
         # torch.dist(spec_lr, spec_ta, p=1)
         # >>> tensor(943.2759)
         assert torch.dist(spec_orig, spec_ta, p=1) < threshold
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_sox_compatibility.py

@@ -299,7 +299,3 @@ def test_perf_biquad_filtering(self):
         data, path = self.get_whitenoise()
         result = F.lfilter(data, torch.tensor([a0, a1, a2]), torch.tensor([b0, b1, b2]))
         self.assert_sox_effect(result, path, ['biquad', b0, b1, b2, a0, a1, a2])
-
-
-if __name__ == "__main__":
-    unittest.main()

test/test_sox_effects.py

@@ -270,7 +270,3 @@ def test_vad(self):
 
             y = vad(x_orig)
             self.assertTrue(x.allclose(y, rtol=1e-4, atol=1e-4))
-
-
-if __name__ == '__main__':
-    unittest.main()

test/test_transforms.py

@@ -219,7 +219,3 @@ def test_compute_deltas_twochannel(self):
         computed = transform(specgram)
         assert computed.shape == expected.shape, (computed.shape, expected.shape)
         self.assertEqual(computed, expected, atol=1e-6, rtol=1e-8)
-
-
-if __name__ == '__main__':
-    unittest.main()