From 44e55c806059aa75df9688af3d90d91c3f8ffd20 Mon Sep 17 00:00:00 2001 From: Tobias Liaudat Date: Mon, 18 Mar 2024 11:43:53 +0100 Subject: [PATCH] apply black formatting --- src/wf_psf/tests/test_utils/utils_test.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/wf_psf/tests/test_utils/utils_test.py b/src/wf_psf/tests/test_utils/utils_test.py index e4717012..fec38f08 100644 --- a/src/wf_psf/tests/test_utils/utils_test.py +++ b/src/wf_psf/tests/test_utils/utils_test.py @@ -36,7 +36,9 @@ def test_unobscured_zernike_projection(): tf_zk_array = tf.convert_to_tensor(zk_array, dtype=tf.float32) # Compute OPD - tf_unobscured_opd = tf.math.reduce_sum(tf.math.multiply(tf_zernike_cube, tf_zk_array), axis=1) + tf_unobscured_opd = tf.math.reduce_sum( + tf.math.multiply(tf_zernike_cube, tf_zk_array), axis=1 + ) # Compute normalisation factor norm_factor = unobscured_zernike_projection( @@ -84,7 +86,9 @@ def test_tf_decompose_obscured_opd_basis(): tf_zk_array = tf.convert_to_tensor(zk_array, dtype=tf.float32) # Compute OPD - tf_unobscured_opd = tf.math.reduce_sum(tf.math.multiply(tf_zernike_cube, tf_zk_array), axis=1) + tf_unobscured_opd = tf.math.reduce_sum( + tf.math.multiply(tf_zernike_cube, tf_zk_array), axis=1 + ) # Obscure the OPD tf_obscured_opd = tf.math.multiply( tf_unobscured_opd, tf.expand_dims(tf_obscurations, axis=0)