Add equality operators for QgsTileXYZ

qgis · May 9, 2023 · 171baf0 · 171baf0
1 parent 35e3d49
commit 171baf0
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/python/core/auto_generated/qgstiles.sip.in b/python/core/auto_generated/qgstiles.sip.in
@@ -47,6 +47,8 @@ Returns tile's zoom level (Z)
 Returns tile coordinates in a formatted string
 %End
 
+    bool operator==( const QgsTileXYZ &other ) const;
+    bool operator!=( const QgsTileXYZ &other ) const;
 
     SIP_PYOBJECT __repr__();
 %MethodCode

diff --git a/src/core/qgstiles.h b/src/core/qgstiles.h
@@ -53,6 +53,9 @@ class CORE_EXPORT QgsTileXYZ
     //! Returns tile coordinates in a formatted string
     QString toString() const { return QStringLiteral( "X=%1 Y=%2 Z=%3" ).arg( mColumn ).arg( mRow ).arg( mZoomLevel ); }
 
+    bool operator==( const QgsTileXYZ &other ) const { return mColumn == other.mColumn && mRow == other.mRow && mZoomLevel == other.mZoomLevel; }
+    bool operator!=( const QgsTileXYZ &other ) const { return !( *this == other ); }
+
 #ifdef SIP_RUN
     SIP_PYOBJECT __repr__();
     % MethodCode

diff --git a/tests/src/python/test_qgstiles.py b/tests/src/python/test_qgstiles.py
@@ -41,6 +41,17 @@ def testQgsTileXYZRepr(self):
         tile = QgsTileXYZ(1, 2, 3)
         self.assertEqual(str(tile), '<QgsTileXYZ: 1, 2, 3>')
 
+    def testQgsTileXYZEquality(self):
+        tile = QgsTileXYZ(1, 2, 3)
+        tile2 = QgsTileXYZ(1, 2, 3)
+        self.assertEqual(tile, tile2)
+        tile2 = QgsTileXYZ(1, 2, 4)
+        self.assertNotEqual(tile, tile2)
+        tile2 = QgsTileXYZ(1, 4, 3)
+        self.assertNotEqual(tile, tile2)
+        tile2 = QgsTileXYZ(4, 2, 3)
+        self.assertNotEqual(tile, tile2)
+
     def testQgsTileRange(self):
         range = QgsTileRange(1, 2, 3, 4)
         self.assertEqual(range.startColumn(), 1)