def test_max_with_initial_arg(self): self.assertQuantityEqual( np.max(self.q[..., None], axis=2, initial=3 * self.ureg.m), [[3, 3], [3, 4]] * self.ureg.m, )
def test_max_with_axis_arg(self): self.assertQuantityEqual(np.max(self.q, axis=1), [2, 4] * self.ureg.m)
def test_max_with_axis_arg(self): helpers.assert_quantity_equal(np.max(self.q, axis=1), [2, 4] * self.ureg.m)
def test_max_numpy_func(self): self.assertEqual(np.max(self.q), 4 * self.ureg.m)
def test_max_numpy_func(self): assert np.max(self.q) == 4 * self.ureg.m