def test_compress_nep18(self): self.assertQuantityEqual(np.compress([False, True], self.q, axis=1), [[2], [4]] * self.ureg.m)
def test_compress_nep18(self): helpers.assert_quantity_equal( np.compress([False, True], self.q, axis=1), [[2], [4]] * self.ureg.m )