commit cfd4768df2e510127235848d4a8cdc5010811bef
author Werner Lewis <werner.lewis@arm.com> Wed Aug 24 17:04:07 2022 +0100
committer Werner Lewis <werner.lewis@arm.com> Wed Aug 24 18:10:13 2022 +0100
tree 03fbec187764272197350711c82953c1da518427
parent 2b527a394dd851905bb73a6ccfd810fbbd7c6d77

Use __new__() for case counting

Signed-off-by: Werner Lewis <werner.lewis@arm.com>

scripts/mbedtls_dev/test_generation.py

diff --git a/scripts/mbedtls_dev/test_generation.py b/scripts/mbedtls_dev/test_generation.py
index aeb551d..f1e085d 100644
--- a/scripts/mbedtls_dev/test_generation.py
+++ b/scripts/mbedtls_dev/test_generation.py
@@ -52,8 +52,9 @@
     test_function = ""
     test_name = ""
 
-    def __init__(self) -> None:
-        type(self).count += 1
+    def __new__(cls, *args, **kwargs):
+        cls.count += 1
+        return super().__new__(cls)
 
     @abstractmethod
     def arguments(self) -> List[str]: