def _test_padding(self, d, k, actual):
def tf_fn():
dt = tf.constant(d, dtype=tf.float32)
kt = tf.constant(k, dtype=tf.float32)
return fft_utils_tf.get_fft_pad_dims(dt, kt)
tf_res = test_utils.exec_tf(tf_fn)
np_res = fft_utils_np.get_fft_pad_dims(d, k)
self.assertTrue(type(tf_res) is np.ndarray)
self.assertTrue(type(np_res) is np.ndarray)
self.assertTrue(np.array_equal(tf_res, np_res))
self.assertTrue(np.array_equal(tf_res, actual))
def _test_optimize_padding(self, d, k, mode, actual):
def tf_fn():
dt = tf.constant(d, dtype=tf.float32)
kt = tf.constant(k, dtype=tf.float32)
pad_dims = fft_utils_tf.get_fft_pad_dims(dt, kt)
return fft_utils_tf.optimize_dims(pad_dims, mode)
tf_res = test_utils.exec_tf(tf_fn)
np_res = fft_utils_np.optimize_dims(fft_utils_np.get_fft_pad_dims(d, k), mode)
self.assertTrue(type(tf_res) is np.ndarray)
self.assertTrue(type(np_res) is np.ndarray)
assert_array_equal(tf_res, np_res)
assert_array_equal(tf_res, actual)