From bda506a2c04a6cf9b2866c7297314b953e626c89 Mon Sep 17 00:00:00 2001
From: Jon Michael Aanes <jon.michael.aanes@partisia.com>
Date: Tue, 15 Apr 2025 10:59:14 +0200
Subject: [PATCH] Support ContractTest

---
 standardize_test_format/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/standardize_test_format/__init__.py b/standardize_test_format/__init__.py
index b8e29a6..b310c20 100644
--- a/standardize_test_format/__init__.py
+++ b/standardize_test_format/__init__.py
@@ -63,7 +63,7 @@ TEST_PATTERN: re.Pattern = re.compile(
     + r'(?:'
     + PATTERN_DISPLAY_NAME_1
     + r'\s*)?'
-    + r'@(?P<annotation>Test|BeforeEach|ParameterizedTest)\s*'
+    + r'@(?P<annotation>Test|BeforeEach|ParameterizedTest|ContractTest(\([^\n]*\))?)\s*'
     + r'(?:'
     + PATTERN_DISPLAY_NAME_2
     + r'\s*)?'