Fix: version test.

Author: ya7010

samples/v1/output.bash

@@ -1,6 +1,6 @@
 #!/bin/bash
 #
-# Code generated by cmdcomp "2.1.0". DO NOT EDIT.
+# Code generated by cmdcomp "$CMDCOMP_VERSION". DO NOT EDIT.
 # For more information about cmdcomp, please refer to https://github.com/yassun7010/cmdcomp .
 #
 

samples/v1/output.zsh

@@ -1,6 +1,6 @@
 #!/bin/zsh
 #
-# Code generated by cmdcomp "2.1.0". DO NOT EDIT.
+# Code generated by cmdcomp "$CMDCOMP_VERSION". DO NOT EDIT.
 # For more information about cmdcomp, please refer to https://github.com/yassun7010/cmdcomp .
 #
 

samples/v2/output.bash

@@ -1,6 +1,6 @@
 #!/bin/bash
 #
-# Code generated by cmdcomp "2.1.0". DO NOT EDIT.
+# Code generated by cmdcomp "$CMDCOMP_VERSION". DO NOT EDIT.
 # For more information about cmdcomp, please refer to https://github.com/yassun7010/cmdcomp .
 #
 

samples/v2/output.zsh

@@ -1,6 +1,6 @@
 #!/bin/zsh
 #
-# Code generated by cmdcomp "2.1.0". DO NOT EDIT.
+# Code generated by cmdcomp "$CMDCOMP_VERSION". DO NOT EDIT.
 # For more information about cmdcomp, please refer to https://github.com/yassun7010/cmdcomp .
 #
 

tests/test_cmdcomp.py

@@ -1,9 +1,7 @@
 import re
 
 import pytest
-from pytest import CaptureFixture
 
-from cmdcomp import __version__
 from cmdcomp.app import App
 from cmdcomp.shell import ShellType
 from tests.conftest import SAMPLES_DIR
@@ -15,7 +13,7 @@
 class TestApplication:
     def test_application(
         self,
-        capsys: CaptureFixture,
+        capsys: pytest.CaptureFixture,
         version: str,
         config_format: str,
         shell: ShellType,
@@ -29,14 +27,15 @@ def test_application(
             ]
         )
 
-        assert capsys.readouterr().out == re.sub(
-            r'\n# Code generated by cmdcomp ".*"\. DO NOT EDIT\.\n',
-            f'\n# Code generated by cmdcomp "{__version__}". DO NOT EDIT.\n',
-            open(SAMPLES_DIR / version / f"output.{str(shell.value)}").read(),
+        assert mask_version(
+            capsys.readouterr().out,
+        ) == mask_version(
+            open(SAMPLES_DIR / version / f"output.{shell.value}").read(),
         )
 
     def test_output_file_to_samples(
         self,
+        capsys: pytest.CaptureFixture,
         version: str,
         config_format: str,
         shell: ShellType,
@@ -47,7 +46,17 @@ def test_output_file_to_samples(
                 str(SAMPLES_DIR / version / f"config.cmdcomp.{config_format}"),
                 "--shell-type",
                 shell.value,
-                "--output",
-                str(SAMPLES_DIR / version / f"output.{shell.value}"),
             ]
         )
+
+        with open(SAMPLES_DIR / version / f"output.{shell.value}", "w") as file:
+            file.write(mask_version(capsys.readouterr().out))
+
+
+def mask_version(output: str) -> str:
+    return re.sub(
+        r'\n# Code generated by cmdcomp ".*"',
+        r'\n# Code generated by cmdcomp "$CMDCOMP_VERSION"',
+        output,
+        count=1,
+    )