code: def test_distributed_strategy(device):
      assert isinstance(device, tf.distribute.Strategy)
      train_small_model()
  
file_path: addons/tensorflow_addons/utils/tests/test_utils_test.py
imports: import random
  from packaging.version import Version
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 4 for calling a global function with example usage: test_distributed_strategy(device) without return types