diff --git a/providers/teradata/tests/unit/teradata/operators/test_bteq.py b/providers/teradata/tests/unit/teradata/operators/test_bteq.py index f943ae09c0828..a099576960f7b 100644 --- a/providers/teradata/tests/unit/teradata/operators/test_bteq.py +++ b/providers/teradata/tests/unit/teradata/operators/test_bteq.py @@ -18,7 +18,6 @@ import logging import tempfile -import unittest from unittest import mock import pytest @@ -288,7 +287,3 @@ def test_bteq_return_code_not_in_quit_rc(self, mock_hook_init, mock_exec): ) result = op.execute({}) assert result == 42 # still returns, but caller can fail on RC if desired - - -if __name__ == "__main__": - unittest.main() diff --git a/providers/teradata/tests/unit/teradata/utils/test_bteq_util.py b/providers/teradata/tests/unit/teradata/utils/test_bteq_util.py index c8a0095f4af6f..0f705c9cf8f7f 100644 --- a/providers/teradata/tests/unit/teradata/utils/test_bteq_util.py +++ b/providers/teradata/tests/unit/teradata/utils/test_bteq_util.py @@ -18,7 +18,6 @@ import os import stat -import unittest from unittest.mock import MagicMock, call, patch import pytest @@ -348,7 +347,3 @@ def test_is_valid_remote_bteq_script_file_none_path(self): ssh_client = MagicMock() result = is_valid_remote_bteq_script_file(ssh_client, None) assert result is False - - -if __name__ == "__main__": - unittest.main() diff --git a/providers/teradata/tests/unit/teradata/utils/test_encryption_utils.py b/providers/teradata/tests/unit/teradata/utils/test_encryption_utils.py index ba688182ae6a9..fdf1c8bc21b82 100644 --- a/providers/teradata/tests/unit/teradata/utils/test_encryption_utils.py +++ b/providers/teradata/tests/unit/teradata/utils/test_encryption_utils.py @@ -17,7 +17,6 @@ from __future__ import annotations import string -import unittest from unittest.mock import MagicMock, patch from airflow.providers.teradata.utils.encryption_utils import ( @@ -102,7 +101,3 @@ def test_decrypt_remote_file_to_string(self): assert exit_status == 0 assert output == "decrypted output" assert err == "" - - -if __name__ == "__main__": - unittest.main()