Skip to content

Commit

Permalink
add tests for collapsing 2 dimensions
Browse files Browse the repository at this point in the history
  • Loading branch information
rcomer committed Aug 30, 2024
1 parent 5d5d6ae commit 638d491
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions lib/iris/tests/unit/coords/test_Coord.py
Original file line number Diff line number Diff line change
Expand Up @@ -738,6 +738,15 @@ def test_string_nd_second(self):

self.assertArrayEqual(collapsed_coord.points, expected)

def test_string_nd_both(self):
self.setupTestArrays((3, 4))
coord = AuxCoord(self.pts_real.astype(str))

collapsed_coord = coord.collapsed()
expected = ["0.0|10.0|20.0|30.0|40.0|50.0|60.0|70.0|80.0|90.0|100.0|110.0"]

self.assertArrayEqual(collapsed_coord.points, expected)

def test_string_nd_bounds_first(self):
self.setupTestArrays((3, 4))
coord = AuxCoord(self.pts_real.astype(str), bounds=self.bds_real.astype(str))
Expand Down Expand Up @@ -784,6 +793,21 @@ def test_string_nd_bounds_second(self):
self.assertArrayEqual(collapsed_coord.bounds[:, 0], expected_lower)
self.assertArrayEqual(collapsed_coord.bounds[:, 1], expected_upper)

def test_string_nd_bounds_both(self):
self.setupTestArrays((3, 4))
coord = AuxCoord(self.pts_real.astype(str), bounds=self.bds_real.astype(str))

collapsed_coord = coord.collapsed()

# Points handling is as for non bounded case. So just check bounds.
expected_lower = ["-2.0|8.0|18.0|28.0|38.0|48.0|58.0|68.0|78.0|88.0|98.0|108.0"]
expected_upper = [
"2.0|12.0|22.0|32.0|42.0|52.0|62.0|72.0|82.0|92.0|102.0|112.0"
]

self.assertArrayEqual(collapsed_coord.bounds[:, 0], expected_lower)
self.assertArrayEqual(collapsed_coord.bounds[:, 1], expected_upper)


class Test_is_compatible(tests.IrisTest):
def setUp(self):
Expand Down

0 comments on commit 638d491

Please sign in to comment.