From: Boyuan Yang Date: Tue, 28 Jan 2025 10:42:25 -0500 Subject: beancount/scripts/doctor_test.py: Allow pytest import --- beancount/scripts/doctor_test.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/beancount/scripts/doctor_test.py b/beancount/scripts/doctor_test.py index 9e7512b..53254c8 100644 --- a/beancount/scripts/doctor_test.py +++ b/beancount/scripts/doctor_test.py @@ -8,7 +8,10 @@ import unittest from os import path from beancount.parser import cmptest -from beancount.scripts import directories_test +try: + from beancount.scripts import directories_test +except ImportError: + from . import directories_test from beancount.scripts.doctor import doctor from beancount.utils import test_utils