def test_nancumprod_numpy_func(self): self.assertRaises(DimensionalityError, np.nancumprod, self.q_nan) self.assertQuantityEqual(np.nancumprod(self.q_nan / self.ureg.m), [1, 2, 6, 6])
def test_nancumprod_numpy_func(self): with pytest.raises(DimensionalityError): np.nancumprod(self.q_nan) helpers.assert_quantity_equal( np.nancumprod(self.q_nan / self.ureg.m), [1, 2, 6, 6] )