Remove unused imports using autoflake

openhtf/__init__.py

@@ -15,7 +15,6 @@
 
 import importlib.metadata
 import signal
-import typing
 
 from openhtf.core import phase_executor
 from openhtf.core import test_record

openhtf/core/base_plugs.py

@@ -94,7 +94,7 @@ def __init__(self, my_config_key)
 """
 
 import logging
-from typing import Any, Dict, Set, Text, Type, Union
+from typing import Any, Dict, Text, Type, Union
 
 import attr
 
@@ -128,17 +128,17 @@ def tearDown(self):
       via TestApi.
   """
   # Override this to True in subclasses to support remote Plug access.
-  enable_remote = False  # type: bool
+  enable_remote: bool = False
   # Allow explicitly disabling remote access to specific attributes.
-  disable_remote_attrs = set()  # type: Set[Text]
+  disable_remote_attrs = set()
   # Override this to True in subclasses to support using with_plugs with this
   # plug without needing to use placeholder.  This will only affect the classes
   # that explicitly define this; subclasses do not share the declaration.
-  auto_placeholder = False  # type: bool
+  auto_placeholder: bool = False
   # Default logger to be used only in __init__ of subclasses.
   # This is overwritten both on the class and the instance so don't store
   # a copy of it anywhere.
-  logger = _LOG  # type: logging.Logger
+  logger: logging.Logger = _LOG
 
   @util.classproperty
   def placeholder(cls) -> 'PlugPlaceholder':  # pylint: disable=no-self-argument
@@ -185,7 +185,7 @@ class FrontendAwareBasePlug(BasePlug, util.SubscribableStateMixin):
   Since the Station API runs in a separate thread, the _asdict() method of
   frontend-aware plugs should be written with thread safety in mind.
   """
-  enable_remote = True  # type: bool
+  enable_remote: bool = True
 
 
 @attr.s(slots=True, frozen=True)

openhtf/core/diagnoses_lib.py

@@ -324,7 +324,6 @@ def possible_results(self) -> List[Text]:
 
   def _check_definition(self) -> None:
     """Internal function to verify that the diagnoser is completely defined."""
-    pass
 
 
 class BasePhaseDiagnoser(_BaseDiagnoser, abc.ABC):

openhtf/core/measurements.py

@@ -63,7 +63,6 @@ def WidgetTestPhase(test):
 import enum
 import functools
 import logging
-import time
 import typing
 from typing import Any, Callable, Dict, Iterator, List, Optional, Text, Tuple, Union
 

openhtf/core/phase_collections.py

@@ -23,7 +23,7 @@
 import abc
 import collections
 from collections.abc import Iterable as CollectionsIterable
-from typing import Any, Callable, DefaultDict, Dict, Iterable, Iterator, List, Optional, Text, Tuple, Type, TypeVar, Union
+from typing import Any, Callable, Dict, Iterable, Iterator, List, Optional, Text, Tuple, Type, TypeVar, Union
 
 import attr
 from openhtf import util
@@ -221,16 +221,18 @@ def check_for_duplicate_subtest_names(sequence: PhaseSequence):
   Raises:
     DuplicateSubtestNamesError: when duplicate subtest names are found.
   """
-  names_to_subtests = collections.defaultdict(
-      list)  # type: DefaultDict[Text, List[Subtest]]
+  names_to_subtests: collections.defaultdict[str, list[Subtest]] = (
+      collections.defaultdict(list)
+  )
   for subtest in sequence.filter_by_type(Subtest):
     names_to_subtests[subtest.name].append(subtest)
 
-  duplicates = []  # type: List[Text]
+  duplicates: list[str] = []
   for name, subtests in names_to_subtests.items():
     if len(subtests) > 1:
-      duplicates.append('Name "{}" used by multiple subtests: {}'.format(
-          name, subtests))
+      duplicates.append(
+          'Name "{}" used by multiple subtests: {}'.format(name, subtests)
+      )
   if not duplicates:
     return
   duplicates.sort()

openhtf/output/callbacks/__init__.py

@@ -80,8 +80,8 @@ class OutputToFile(object):
 
   def __init__(self, filename_pattern_or_file: Union[Text, Callable[..., Text],
                                                      BinaryIO]):
-    self.filename_pattern = None  # type: Optional[Union[Text, Callable[..., Text]]]
-    self.output_file = None  # type: Optional[BinaryIO]
+    self.filename_pattern: Optional[Union[Text, Callable[..., Text]]] = None
+    self.output_file: Optional[BinaryIO] = None
     if (isinstance(filename_pattern_or_file, str) or
         callable(filename_pattern_or_file)):
       self.filename_pattern = filename_pattern_or_file  # pytype: disable=annotation-type-mismatch

openhtf/output/servers/pub_sub.py

@@ -67,8 +67,6 @@ def on_close(self):
 
   def on_subscribe(self, info):
     """Called when new clients subscribe. Subclasses can override."""
-    pass
 
   def on_unsubscribe(self):
     """Called when clients unsubscribe. Subclasses can override."""
-    pass

openhtf/plugs/generic/serial_collection.py

@@ -58,10 +58,10 @@ class SerialCollectionPlug(base_plugs.BasePlug):
   # Serial library can raise these exceptions
   SERIAL_EXCEPTIONS = (serial.SerialException, ValueError)
 
-  _serial = None  # type: serial.Serial
-  _serial_port = None  # type: int
-  _collect = None  # type: bool
-  _collection_thread = None  # type: Optional[threading.Thread]
+  _serial: Optional[serial.Serial] = None
+  _serial_port: Optional[int] = None
+  _collect: Optional[bool] = None
+  _collection_thread: Optional[threading.Thread] = None
 
   @CONF.inject_positional_args
   def __init__(self, serial_collection_port, serial_collection_baud):

openhtf/plugs/user_input.py

@@ -24,7 +24,7 @@
 import select
 import sys
 import threading
-from typing import Any, Callable, Dict, Optional, Text, Tuple, Union
+from typing import Any, Callable, Dict, Optional, Text, Union
 import uuid
 
 import attr
@@ -142,10 +142,10 @@ class UserInput(base_plugs.FrontendAwareBasePlug):
 
   def __init__(self):
     super(UserInput, self).__init__()
-    self.last_response = None  # type: Optional[Tuple[Text, Text]]
-    self._prompt = None  # type: Optional[Prompt]
-    self._console_prompt = None  # type: Optional[ConsolePrompt]
-    self._response = None  # type: Optional[Text]
+    self.last_response: Optional[tuple[str, str]] = None
+    self._prompt: Optional[Prompt] = None
+    self._console_prompt: Optional[ConsolePrompt] = None
+    self._response: Optional[Text] = None
     self._cond = threading.Condition(threading.RLock())
 
   def _asdict(self) -> Optional[Dict[Text, Any]]:

openhtf/util/__init__.py

@@ -18,7 +18,7 @@
 import threading
 import time
 import typing
-from typing import Any, Callable, Dict, Iterator, Optional, Text, Tuple, Type, TypeVar, Union
+from typing import Any, Callable, Dict, Iterator, Optional, Text, Tuple, TypeVar
 import weakref
 
 import attr

openhtf/util/logs.py

@@ -110,7 +110,6 @@ def MyPhase(test, helper):
 from openhtf.util import argv
 from openhtf.util import console_output
 from openhtf.util import functions
-from openhtf.util import threads
 
 # The number of v's provided as command line arguments to control verbosity.
 # Will be overridden if the ARG_PARSER below parses the -v argument.

openhtf/util/test.py

@@ -126,7 +126,6 @@ def test_multiple(self, mock_my_plug):
   assertMeasurementFail(phase_or_test_rec, measurement)
 """
 
-import collections
 from collections.abc import Callable as CollectionsCallable, Iterator
 import functools
 import inspect

openhtf/util/threads.py

@@ -13,8 +13,6 @@
 # limitations under the License.
 """Thread library defining a few helpers."""
 
-import _thread
-import contextlib
 import cProfile
 import ctypes
 import functools

openhtf/util/validators.py

@@ -71,7 +71,7 @@ def MyPhase(test: htf.TestApi) -> None:
 import math
 import numbers
 import re
-from typing import Callable, Dict, Optional, Type, TypeVar, Union
+from typing import Callable, Dict, Optional, Union
 
 from openhtf import util
 

openhtf/util/xmlrpcutil.py

@@ -16,7 +16,6 @@
 from collections.abc import Callable
 import http.client
 import socketserver
-import sys
 import threading
 import xmlrpc.client
 import xmlrpc.server

test/core/measurements_test.py

@@ -19,7 +19,6 @@
 """
 
 import collections
-import unittest
 from unittest import mock
 
 import openhtf as htf

test/output/callbacks/callbacks_test.py

@@ -20,7 +20,6 @@
 
 import io
 import json
-import unittest
 
 import openhtf as htf
 from openhtf import util

test/output/callbacks/mfg_inspector_test.py

@@ -19,7 +19,6 @@
 """
 import collections
 import io
-import unittest
 from unittest import mock
 
 import openhtf as htf

test/phase_descriptor_test.py

@@ -30,7 +30,6 @@
 
 def plain_func():
   """Plain Docstring."""
-  pass
 
 
 def normal_test_phase():

test/plugs/plugs_test.py

@@ -14,7 +14,6 @@
 
 import threading
 import time
-import unittest
 
 from openhtf import plugs
 from openhtf.core import base_plugs

test/test_state_test.py

@@ -16,7 +16,6 @@
 import logging
 import sys
 import tempfile
-import unittest
 from unittest import mock
 
 from absl.testing import parameterized
@@ -37,7 +36,6 @@
 @openhtf.PhaseOptions(name='test_phase')
 def test_phase():
   """Test docstring."""
-  pass
 
 
 PHASE_STATE_BASE_TYPE_INITIAL = {

test/util/text_test.py

@@ -17,7 +17,6 @@
 import sys
 import types
 import typing
-import unittest
 from unittest import mock
 
 from absl.testing import parameterized