diff options
| -rw-r--r-- | numpy/core/tests/test_defchararray.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/numpy/core/tests/test_defchararray.py b/numpy/core/tests/test_defchararray.py index 2e8db9e9a..e1e0d677c 100644 --- a/numpy/core/tests/test_defchararray.py +++ b/numpy/core/tests/test_defchararray.py @@ -41,10 +41,12 @@ class TestOperations(TestCase): assert all(QA == ('q' + self.A)) def test_mul(self): - A2 = np.array([['abcabc', '123123'], - ['789789', 'xyzxyz']]).view(np.chararray) + A = self.A + for r in (2,3,5,7,197): + Ar = np.array([[A[0,0]*r, A[0,1]*r], + [A[1,0]*r, A[1,1]*r]]).view(np.chararray) - assert all(A2 == (self.A * 2)) + assert all(Ar == (self.A * r)) for ob in [object(), 'qrs']: try: @@ -55,10 +57,12 @@ class TestOperations(TestCase): self.fail("chararray can only be multiplied by integers") def test_rmul(self): - A2 = np.array([['abcabc', '123123'], - ['789789', 'xyzxyz']]).view(np.chararray) + A = self.A + for r in (2,3,5,7,197): + Ar = np.array([[A[0,0]*r, A[0,1]*r], + [A[1,0]*r, A[1,1]*r]]).view(np.chararray) - assert all(A2 == (2 * self.A)) + assert all(Ar == (r * self.A)) for ob in [object(), 'qrs']: try: |