code: def test_diag_scale():
      x0 = np.asarray([[0.1, 0.2, 0.3]]).astype(np.float32)
      x = np.asarray([[0.4, 0.5, 0.6]]).astype(np.float32)
      layer = PolynomialCrossing(
          projection_dim=None, diag_scale=1.0, kernel_initializer="ones"
      )
      output = layer([x0, x])
      np.testing.assert_allclose([[0.59, 0.9, 1.23]], output)
  
file_path: addons/tensorflow_addons/layers/tests/polynomial_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.layers.polynomial import PolynomialCrossing
label_desc: Use image node_id 6 for calling a global function with example usage: test_diag_scale() without return types