Commits

David Mehringer authored 2c5489dabd5
more debug info for statwt bamboo failures on gpu

casatools/tests/tools/ms/test_tool_ms_statwt.py

Modified
248 248 [
249 249 gtimes, gwt, gwtsp, gflag, gfrow, gdata, gsigma, gsisp
250 250 ] = _get_table_cols(mytb)
251 251 mytb.done()
252 252 ref = os.path.join(refdir, ref)
253 253 self.assertTrue(mytb.open(ref), "Table open failed for " + ref)
254 254 [
255 255 etimes, ewt, ewtsp, eflag, efrow, edata, esigma, esisp
256 256 ] = _get_table_cols(mytb)
257 257 mytb.done()
258 - self.assertTrue(np.allclose(gwt, ewt), 'WEIGHT comparison failed')
258 + self.assertTrue(
259 + np.allclose(gwt, ewt),
260 + f'WEIGHT comparison failed. max diff {np.max(np.abs(gwt - ewt))}'
261 + )
259 262 if type(gwtsp) != type(None) and type(ewtsp) != type(None):
260 263 self.assertTrue(
261 264 np.allclose(gwtsp, ewtsp), 'WEIGHT_SPECTRUM comparison failed'
262 265 f"max diff {np.max(np.abs(gwtsp - ewtsp))}"
263 266 )
264 267 self.assertTrue((gflag == eflag).all(), 'FLAG comparison failed')
265 268 self.assertTrue((gfrow == efrow).all(), 'FLAG_ROW comparison failed')
266 269 # all flags must be True where wtsp = 0
267 270 self.assertTrue(np.extract(gwtsp == 0, gflag).all())
268 271 self.assertTrue(np.allclose(gsigma, esigma), 'SIGMA comparison failed')

Everything looks good. We'll let you know here if there's anything you should know about.

Add shortcut