From 10112f3c92e4b22a36d9e492ec736197942bb6ae Mon Sep 17 00:00:00 2001 From: Vitaliy Mysak Date: Fri, 20 Sep 2024 14:26:37 -0700 Subject: [PATCH] Add 1 more test case for CoordinateMapping.__repr__ --- tests/test_main.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/test_main.py b/tests/test_main.py index 45d8517..500cd09 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1010,6 +1010,13 @@ def test_coordinate_mapping_repr(): mapping_4.extend(2, 3, 1) mapping_4.extend(3, 4, 2) + mapping_5 = CoordinateMapping() + mapping_5.extend(2, 3, 1) + mapping_5.extend(3, 4, 2) + mapping_5.extend(None, 4, 3) + mapping_5.extend(4, None, 4) + mapping_5.extend(5, 5, 5) + assert repr(mapping_1) \ == repr(mapping_2) \ == "CoordinateMapping({ 2: 0, 3: 1 }, { 3: 0, 4: 1 })" @@ -1019,3 +1026,10 @@ def test_coordinate_mapping_repr(): assert repr(mapping_4) \ == "CoordinateMapping({ 2: 1, 3: 2 }, { 3: 1, 4: 2 })" + + assert repr(mapping_5) \ + == "CoordinateMapping(" \ + "{ 2: 1, 3: 2, 4: 4, 5: 5, None: 3 }" \ + ", " \ + "{ 3: 1, 4: 3, 5: 5, None: 2, None: 4 }" \ + ")"