From 86409f0de0c65f8e44602ad8dcbae9a8044b0513 Mon Sep 17 00:00:00 2001 From: Alan McIntyre Date: Thu, 10 Jul 2008 12:47:28 +0000 Subject: Improve chararray __mul__/__rmul__ test so it should fail on 64-bit machines. --- numpy/core/tests/test_defchararray.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (limited to 'numpy') 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: -- cgit v1.2.1