From 9a90abf995d0d8d9e96992a083dc55a41a93254f Mon Sep 17 00:00:00 2001 From: MattHarrigan Date: Thu, 27 Oct 2016 19:38:01 -0400 Subject: BUG: return subclasses from ediff1d --- numpy/lib/tests/test_arraysetops.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'numpy/lib/tests/test_arraysetops.py') diff --git a/numpy/lib/tests/test_arraysetops.py b/numpy/lib/tests/test_arraysetops.py index b75a2b060..75918fbee 100644 --- a/numpy/lib/tests/test_arraysetops.py +++ b/numpy/lib/tests/test_arraysetops.py @@ -175,6 +175,8 @@ class TestSetOps(TestCase): assert_array_equal([1,7,8], ediff1d(two_elem, to_end=[7,8])) assert_array_equal([7,1], ediff1d(two_elem, to_begin=7)) assert_array_equal([5,6,1], ediff1d(two_elem, to_begin=[5,6])) + assert(isinstance(ediff1d(np.matrix(1)), np.matrix)) + assert(isinstance(ediff1d(np.matrix(1), to_begin=1), np.matrix)) def test_in1d(self): # we use two different sizes for the b array here to test the -- cgit v1.2.1