diff --git a/hera_filters/tests/test_dspec.py b/hera_filters/tests/test_dspec.py index 5c7c3f8..0764766 100644 --- a/hera_filters/tests/test_dspec.py +++ b/hera_filters/tests/test_dspec.py @@ -1512,6 +1512,7 @@ def test_sparse_linear_fit_2d_non_binary_wgts(): data = np.dot(time_basis, x_true).dot(freq_basis.T) flags = (time_flags | freq_flags) + # Generate separable, non-binary weights axis_1_weights = (~time_flags[:, 0]).astype(float) * np.random.randint(1, 10, size=(ntimes, )) axis_2_weights = (~freq_flags[0]).astype(float) wgts = np.outer(axis_1_weights, axis_2_weights)