From 580da1194931a47a83e3872511d3e78b87d6a75d Mon Sep 17 00:00:00 2001
From: Isaiah Odhner <isaiahodhner@gmail.com>
Date: Thu, 7 Sep 2023 15:30:49 -0400
Subject: [PATCH] Check if running in pytest instead of ignoring a specific
 argument

---
 README.md                 | 2 --
 src/textual_paint/args.py | 7 ++++---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 20ca7e0..55a0e14 100644
--- a/README.md
+++ b/README.md
@@ -88,8 +88,6 @@ development options:
                         outdated errors
   --restart-on-changes  Restart the app when the source code is changed
   --recode-samples      Open and save each file in samples/, for testing
-  --snapshot-update     KLUDGE: IGNORE THIS ARGUMENT INTENDED FOR pytest-
-                        textual-snapshot
 ```
 
 ### Keyboard Shortcuts
diff --git a/src/textual_paint/args.py b/src/textual_paint/args.py
index e6e1daf..ca5d525 100644
--- a/src/textual_paint/args.py
+++ b/src/textual_paint/args.py
@@ -3,9 +3,12 @@
 import argparse
 import os
 import re
+import sys
 
 from textual_paint.__init__ import DEVELOPMENT, __version__
 
+PYTEST = "pytest" in sys.modules
+
 parser = argparse.ArgumentParser(description='Paint in the terminal.', usage='%(prog)s [options] [filename]', prog="textual-paint")
 parser.add_argument('--version', action='version', version=f'%(prog)s {__version__}')
 parser.add_argument('--theme', default='light', help='Theme to use, either "light" or "dark"', choices=['light', 'dark'])
@@ -25,8 +28,6 @@ dev_options.add_argument('--inspect-layout', action='store_true', help='Enables
 dev_options.add_argument('--clear-screen', action='store_true', help='Clear the screen before starting, to avoid seeing outdated errors')
 dev_options.add_argument('--restart-on-changes', action='store_true', help='Restart the app when the source code is changed')
 dev_options.add_argument('--recode-samples', action='store_true', help='Open and save each file in samples/, for testing')
-# TODO: don't parse arguments when running tests!
-dev_options.add_argument('--snapshot-update', action='store_true', help='KLUDGE: IGNORE THIS ARGUMENT INTENDED FOR pytest-textual-snapshot')
 
 parser.add_argument('filename', nargs='?', default=None, help='Path to a file to open. File will be created if it doesn\'t exist.')
 
@@ -71,7 +72,7 @@ if DEVELOPMENT:
     # while working on the project.
     update_cli_help_on_readme()
 
-args = parser.parse_args()
+args = parser.parse_args([]) if PYTEST else parser.parse_args()
 """Parsed command line arguments."""
 
 def get_help_text() -> str: