From 0cfdf55c3729e5615cf9daa6fc805f49b1393db7 Mon Sep 17 00:00:00 2001 From: Simon Perkins Date: Mon, 25 Mar 2024 16:56:29 +0200 Subject: [PATCH] Move test_transpose precondition --- daskms/experimental/katdal/tests/test_chunkstore.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/daskms/experimental/katdal/tests/test_chunkstore.py b/daskms/experimental/katdal/tests/test_chunkstore.py index cfd7c21e..f5ddcd31 100644 --- a/daskms/experimental/katdal/tests/test_chunkstore.py +++ b/daskms/experimental/katdal/tests/test_chunkstore.py @@ -289,11 +289,12 @@ def test_chunkstore(tmp_path_factory, dataset, include_auto_corrs, row_dim, out_ (nbl,) = cp_info.ant1_index.shape ncorr = read_xds.sizes["corr"] + # This must hold for test_tranpose to work + assert_array_equal(cp_info.cp_index.ravel(), np.arange(nbl * ncorr)) + def test_transpose(a): """Simple transpose of katdal (time, chan, corrprod) to (time, bl, chan, corr).""" - # This must hold for this simple tranpose to work - assert_array_equal(cp_info.cp_index.ravel(), np.arange(nbl * ncorr)) o = a.reshape(ntime, nchan, nbl, ncorr).transpose(0, 2, 1, 3) return o.reshape(-1, nchan, ncorr) if row_dim else o