diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2020-09-07 16:30:21 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-09-07 16:30:21 (GMT) |
commit | 1671b0e4e00061f23b6677e37e27a4ed44998329 (patch) | |
tree | de1e4b54fc1ebff84bef1b935e399b996ebeed14 /Lib/test/test_turtle.py | |
parent | c73ee5acc96b4bbd6885156883b224b8cc3e470c (diff) | |
download | cpython-1671b0e4e00061f23b6677e37e27a4ed44998329.zip cpython-1671b0e4e00061f23b6677e37e27a4ed44998329.tar.gz cpython-1671b0e4e00061f23b6677e37e27a4ed44998329.tar.bz2 |
bpo-41720: Add "return NotImplemented" in turtle.Vec2D.__rmul__(). (GH-22092)
(cherry picked from commit fd4cafd4700dc03cb05fc2e5263c2666d785d6e3)
Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
Diffstat (limited to 'Lib/test/test_turtle.py')
-rw-r--r-- | Lib/test/test_turtle.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/Lib/test/test_turtle.py b/Lib/test/test_turtle.py index 38448c7..5d5f057 100644 --- a/Lib/test/test_turtle.py +++ b/Lib/test/test_turtle.py @@ -127,6 +127,14 @@ class VectorComparisonMixin: self.assertAlmostEqual( i, j, msg='values at index {} do not match'.format(idx)) +class Multiplier: + + def __mul__(self, other): + return f'M*{other}' + + def __rmul__(self, other): + return f'{other}*M' + class TestVec2D(VectorComparisonMixin, unittest.TestCase): @@ -208,9 +216,15 @@ class TestVec2D(VectorComparisonMixin, unittest.TestCase): self.assertAlmostEqual(answer, expected) vec = Vec2D(0.5, 3) - answer = vec * 10 expected = Vec2D(5, 30) - self.assertVectorsAlmostEqual(answer, expected) + self.assertVectorsAlmostEqual(vec * 10, expected) + self.assertVectorsAlmostEqual(10 * vec, expected) + self.assertVectorsAlmostEqual(vec * 10.0, expected) + self.assertVectorsAlmostEqual(10.0 * vec, expected) + + M = Multiplier() + self.assertEqual(vec * M, Vec2D(f"{vec[0]}*M", f"{vec[1]}*M")) + self.assertEqual(M * vec, f'M*{vec}') def test_vector_negative(self): vec = Vec2D(10, -10) |