1 files changed, 92 insertions, 0 deletions
diff --git a/sources/pyside6/tests/QtCore/qbytearray_operator_test.py b/sources/pyside6/tests/QtCore/qbytearray_operator_test.py
new file mode 100644
index 000000000..0064e4dd6
--- /dev/null
+++ b/sources/pyside6/tests/QtCore/qbytearray_operator_test.py
@@ -0,0 +1,92 @@
+#!/usr/bin/python
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+'''Test cases for QByteArray operators'''
+
+import os
+import sys
+import unittest
+
+from pathlib import Path
+sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
+from init_paths import init_test_paths
+init_test_paths(False)
+
+from PySide6.QtCore import QByteArray
+
+
+class QByteArrayOperatorEqual(unittest.TestCase):
+    '''TestCase for operator QByteArray == QByteArray'''
+
+    def testDefault(self):
+        # QByteArray() == QByteArray()
+        obj1 = QByteArray()
+        obj2 = QByteArray()
+        self.assertEqual(obj1, obj2)
+
+    def testSimple(self):
+        # QByteArray(some_string) == QByteArray(some_string)
+        string = bytes('egg snakes', "UTF-8")
+        self.assertEqual(QByteArray(string), QByteArray(string))
+
+    def testPyString(self):
+        # QByteArray(string) == string
+        string = bytes('my test string', "UTF-8")
+        self.assertEqual(QByteArray(string), string)
+
+
+class QByteArrayOperatorAt(unittest.TestCase):
+    '''TestCase for operator QByteArray[]'''
+
+    def testInRange(self):
+        # QByteArray[x] where x is a valid index
+        string = 'abcdefgh'
+        obj = QByteArray(bytes(string, "UTF-8"))
+
+        for i in range(len(string)):
+            self.assertEqual(obj[i], bytes(string[i], "UTF-8"))
+
+    def testInRangeReverse(self):
+        # QByteArray[x] where x is a valid index (reverse order)
+        string = 'abcdefgh'
+        obj = QByteArray(bytes(string, "UTF-8"))
+
+        for i in range(len(string) - 1, 0, -1):
+            self.assertEqual(obj[i], bytes(string[i], "UTF-8"))
+
+    def testOutOfRange(self):
+        # QByteArray[x] where x is out of index
+        string = bytes('1234567', "UTF-8")
+        obj = QByteArray(string)
+        self.assertRaises(IndexError, lambda: obj[len(string)])
+
+    def testNullStrings(self):
+        ba = QByteArray(bytes('\x00', "UTF-8"))
+        self.assertEqual(ba.at(0), '\x00')
+        self.assertEqual(ba[0], bytes('\x00', "UTF-8"))
+
+
+class QByteArrayOperatorLen(unittest.TestCase):
+    '''Test case for __len__ operator of QByteArray'''
+
+    def testBasic(self):
+        '''QByteArray __len__'''
+        self.assertEqual(len(QByteArray()), 0)
+        self.assertEqual(len(QByteArray(bytes('', "UTF-8"))), 0)
+        self.assertEqual(len(QByteArray(bytes(' ', "UTF-8"))), 1)
+        self.assertEqual(len(QByteArray(bytes('yabadaba', "UTF-8"))), 8)
+
+
+class QByteArrayOperatorStr(unittest.TestCase):
+    '''Test case for __str__ operator of QByteArray'''
+
+    def testBasic(self):
+        '''QByteArray __str__'''
+        self.assertEqual(QByteArray().__str__(), str(b''))
+        self.assertEqual(QByteArray(bytes('', "UTF-8")).__str__(), str(b''))
+        self.assertEqual(QByteArray(bytes('aaa', "UTF-8")).__str__(), str(b'aaa'))
+
+
+if __name__ == '__main__':
+    unittest.main()