From 7c3171619f2c5f3cf1667de1b19724176f5dc479 Mon Sep 17 00:00:00 2001
From: Laurent Monin <github@norz.org>
Date: Sat, 20 Nov 2021 13:02:05 +0100
Subject: [PATCH] Add tests for _regex_numbered_title_fmt()

---
 test/test_util_uniqnum_title.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/test/test_util_uniqnum_title.py b/test/test_util_uniqnum_title.py
index fa5ab9ce0..f05f19520 100644
--- a/test/test_util_uniqnum_title.py
+++ b/test/test_util_uniqnum_title.py
@@ -21,7 +21,33 @@
 
 from test.picardtestcase import PicardTestCase
 
-from picard.util import unique_numbered_title
+from picard.util import (
+    _regex_numbered_title_fmt,
+    unique_numbered_title,
+)
+
+
+class RegexNumberedTitleFmt(PicardTestCase):
+
+    def test_1(self):
+        fmt = ''
+        result = _regex_numbered_title_fmt(fmt, 'TITLE', 'COUNT')
+        self.assertEqual(result, '')
+
+    def test_2(self):
+        fmt = '{title} {count}'
+        result = _regex_numbered_title_fmt(fmt, 'TITLE', 'COUNT')
+        self.assertEqual(result, r'TITLE(?:\ COUNT)?')
+
+    def test_3(self):
+        fmt = 'x {count}  {title} y'
+        result = _regex_numbered_title_fmt(fmt, 'TITLE', 'COUNT')
+        self.assertEqual(result, r'(?:x\ COUNT\ \ )?TITLE y')
+
+    def test_4(self):
+        fmt = 'x {title}{count} y'
+        result = _regex_numbered_title_fmt(fmt, 'TITLE', 'COUNT')
+        self.assertEqual(result, r'x TITLE(?:COUNT\ y)?')
 
 
 class UniqueNumberedTitle(PicardTestCase):