def test_diagonal_numpy_func(self):
q = [[1, 2, 3], [1, 2, 3], [1, 2, 3]] * self.ureg.m
self.assertQuantityEqual(np.diagonal(q, offset=-1),
[1, 2] * self.ureg.m)
def test_diagonal_numpy_func(self):
q = [[1, 2, 3], [1, 2, 3], [1, 2, 3]] * self.ureg.m
helpers.assert_quantity_equal(np.diagonal(q, offset=-1), [1, 2] * self.ureg.m)
