def test_cumprod_numpy_func(self): self.assertRaises(DimensionalityError, np.cumprod, self.q) self.assertRaises(DimensionalityError, np.cumproduct, self.q) self.assertQuantityEqual(np.cumprod(self.q / self.ureg.m), [1, 2, 6, 24]) self.assertQuantityEqual(np.cumproduct(self.q / self.ureg.m), [1, 2, 6, 24]) self.assertQuantityEqual( np.cumprod(self.q / self.ureg.m, axis=1), [[1, 2], [3, 12]] )
def test_cumprod_numpy_func(self): with pytest.raises(DimensionalityError): np.cumprod(self.q) with pytest.raises(DimensionalityError): np.cumproduct(self.q) helpers.assert_quantity_equal(np.cumprod(self.q / self.ureg.m), [1, 2, 6, 24]) helpers.assert_quantity_equal( np.cumproduct(self.q / self.ureg.m), [1, 2, 6, 24] ) helpers.assert_quantity_equal( np.cumprod(self.q / self.ureg.m, axis=1), [[1, 2], [3, 12]] )