Standardized Testing Module

tensorflow_addons/optimizers/tests/standard_test.py

@@ -18,7 +18,7 @@
 import tensorflow as tf
 
 from tensorflow_addons import optimizers
-import inspect
+from tensorflow_addons.utils.test_utils import discover_classes
 
 class_exceptions = [
     "MultiOptimizer",  # is wrapper
@@ -32,18 +32,9 @@
 ]
 
 
-def discover_classes(module, parent):
-
-    classes = [
-        class_info[1]
-        for class_info in inspect.getmembers(module, inspect.isclass)
-        if issubclass(class_info[1], parent) and not class_info[0] in class_exceptions
-    ]
-
-    return classes
-
-
-classes_to_test = discover_classes(optimizers, tf.keras.optimizers.Optimizer)
+classes_to_test = discover_classes(
+    optimizers, tf.keras.optimizers.Optimizer, class_exceptions
+)
 
 
 @pytest.mark.parametrize("optimizer", classes_to_test)

tensorflow_addons/utils/test_utils.py

@@ -16,6 +16,7 @@
 
 import os
 import random
+import inspect
 
 import numpy as np
 import pytest
@@ -271,3 +272,23 @@ def assert_allclose_according_to_type(
         atol = max(atol, bfloat16_atol)
 
     np.testing.assert_allclose(a, b, rtol=rtol, atol=atol)
+
+
+def discover_classes(module, parent, class_exceptions):
+    """
+    Args:
+        module: a module in which to search for classes that inherit from the parent class
+        parent: the parent class that identifies classes in the module that should be tested
+        class_exceptions: a list of specific classes that should be excluded when discovering classes in a module
+
+    Returns:
+        a list of classes for testing using pytest for parameterized tests
+    """
+
+    classes = [
+        class_info[1]
+        for class_info in inspect.getmembers(module, inspect.isclass)
+        if issubclass(class_info[1], parent) and not class_info[0] in class_exceptions
+    ]
+
+    return classes