Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
59dfe084
Commit
59dfe084
authored
Oct 30, 2022
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
launch tests from launch.py with --tests commandline argument
parent
05a657dd
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
50 additions
and
40 deletions
+50
-40
.gitignore
.gitignore
+2
-0
launch.py
launch.py
+19
-0
run_tests.bat
run_tests.bat
+0
-15
extras_test.py
test/extras_test.py
+3
-3
img2img_test.py
test/img2img_test.py
+6
-6
server_poll.py
test/server_poll.py
+15
-13
txt2img_test.py
test/txt2img_test.py
+5
-3
No files found.
.gitignore
View file @
59dfe084
...
@@ -29,3 +29,5 @@ notification.mp3
...
@@ -29,3 +29,5 @@ notification.mp3
/textual_inversion
/textual_inversion
.vscode
.vscode
/extensions
/extensions
/test/stdout.txt
/test/stderr.txt
launch.py
View file @
59dfe084
...
@@ -128,10 +128,12 @@ def prepare_enviroment():
...
@@ -128,10 +128,12 @@ def prepare_enviroment():
blip_commit_hash
=
os
.
environ
.
get
(
'BLIP_COMMIT_HASH'
,
"48211a1594f1321b00f14c9f7a5b4813144b2fb9"
)
blip_commit_hash
=
os
.
environ
.
get
(
'BLIP_COMMIT_HASH'
,
"48211a1594f1321b00f14c9f7a5b4813144b2fb9"
)
sys
.
argv
+=
shlex
.
split
(
commandline_args
)
sys
.
argv
+=
shlex
.
split
(
commandline_args
)
test_argv
=
[
x
for
x
in
sys
.
argv
if
x
!=
'--tests'
]
sys
.
argv
,
skip_torch_cuda_test
=
extract_arg
(
sys
.
argv
,
'--skip-torch-cuda-test'
)
sys
.
argv
,
skip_torch_cuda_test
=
extract_arg
(
sys
.
argv
,
'--skip-torch-cuda-test'
)
sys
.
argv
,
reinstall_xformers
=
extract_arg
(
sys
.
argv
,
'--reinstall-xformers'
)
sys
.
argv
,
reinstall_xformers
=
extract_arg
(
sys
.
argv
,
'--reinstall-xformers'
)
sys
.
argv
,
update_check
=
extract_arg
(
sys
.
argv
,
'--update-check'
)
sys
.
argv
,
update_check
=
extract_arg
(
sys
.
argv
,
'--update-check'
)
sys
.
argv
,
run_tests
=
extract_arg
(
sys
.
argv
,
'--tests'
)
xformers
=
'--xformers'
in
sys
.
argv
xformers
=
'--xformers'
in
sys
.
argv
deepdanbooru
=
'--deepdanbooru'
in
sys
.
argv
deepdanbooru
=
'--deepdanbooru'
in
sys
.
argv
ngrok
=
'--ngrok'
in
sys
.
argv
ngrok
=
'--ngrok'
in
sys
.
argv
...
@@ -194,6 +196,23 @@ def prepare_enviroment():
...
@@ -194,6 +196,23 @@ def prepare_enviroment():
print
(
"Exiting because of --exit argument"
)
print
(
"Exiting because of --exit argument"
)
exit
(
0
)
exit
(
0
)
if
run_tests
:
tests
(
test_argv
)
exit
(
0
)
def
tests
(
argv
):
print
(
f
"Launching Web UI in another process for testing with arguments: {' '.join(argv[1:])}"
)
with
open
(
'test/stdout.txt'
,
"w"
,
encoding
=
"utf8"
)
as
stdout
,
open
(
'test/stderr.txt'
,
"w"
,
encoding
=
"utf8"
)
as
stderr
:
proc
=
subprocess
.
Popen
([
sys
.
executable
,
*
argv
],
stdout
=
stdout
,
stderr
=
stderr
)
import
test.server_poll
test
.
server_poll
.
run_tests
()
print
(
f
"Stopping Web UI process with id {proc.pid}"
)
proc
.
kill
()
def
start_webui
():
def
start_webui
():
print
(
f
"Launching Web UI with arguments: {' '.join(sys.argv[1:])}"
)
print
(
f
"Launching Web UI with arguments: {' '.join(sys.argv[1:])}"
)
...
...
run_tests.bat
deleted
100644 → 0
View file @
05a657dd
@echo off
set ERROR_REPORTING=FALSE
set COMMANDLINE_ARGS= --api
echo Launching SDWebUI...
start "SDWebUITest" webui.bat
if not defined PYTHON (set PYTHON=python)
if not defined VENV_DIR (set VENV_DIR=venv)
set PYTHON="%~dp0%VENV_DIR%\Scripts\Python.exe"
%PYTHON% test/server_poll.py
for /f "tokens=2 delims=," %%a in ('tasklist /v /fo csv ^| findstr /i "SDWebUITest"') do set "$PID=%%a"
taskkill /PID %$PID% >nul 2>&1
pause
test/extras_test.py
View file @
59dfe084
import
unittest
import
unittest
import
requests
from
gradio.processing_utils
import
encode_pil_to_base64
from
PIL
import
Image
class
TestExtrasWorking
(
unittest
.
TestCase
):
class
TestExtrasWorking
(
unittest
.
TestCase
):
def
setUp
(
self
):
def
setUp
(
self
):
...
@@ -22,8 +20,10 @@ class TestExtrasWorking(unittest.TestCase):
...
@@ -22,8 +20,10 @@ class TestExtrasWorking(unittest.TestCase):
"image"
:
""
"image"
:
""
}
}
class
TestExtrasCorrectness
(
unittest
.
TestCase
):
class
TestExtrasCorrectness
(
unittest
.
TestCase
):
pass
pass
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
unittest
.
main
()
unittest
.
main
()
test/img2img_test.py
View file @
59dfe084
...
@@ -3,13 +3,12 @@ import requests
...
@@ -3,13 +3,12 @@ import requests
from
gradio.processing_utils
import
encode_pil_to_base64
from
gradio.processing_utils
import
encode_pil_to_base64
from
PIL
import
Image
from
PIL
import
Image
class
TestImg2ImgWorking
(
unittest
.
TestCase
):
class
TestImg2ImgWorking
(
unittest
.
TestCase
):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
url_img2img
=
"http://localhost:7860/sdapi/v1/img2img"
self
.
url_img2img
=
"http://localhost:7860/sdapi/v1/img2img"
self
.
simple_img2img
=
{
self
.
simple_img2img
=
{
"init_images"
:
[
"init_images"
:
[
encode_pil_to_base64
(
Image
.
open
(
r"test/test_files/img2img_basic.png"
))],
encode_pil_to_base64
(
Image
.
open
(
r"test/test_files/img2img_basic.png"
))
],
"resize_mode"
:
0
,
"resize_mode"
:
0
,
"denoising_strength"
:
0.75
,
"denoising_strength"
:
0.75
,
"mask"
:
None
,
"mask"
:
None
,
...
@@ -19,9 +18,7 @@ class TestImg2ImgWorking(unittest.TestCase):
...
@@ -19,9 +18,7 @@ class TestImg2ImgWorking(unittest.TestCase):
"inpaint_full_res_padding"
:
0
,
"inpaint_full_res_padding"
:
0
,
"inpainting_mask_invert"
:
0
,
"inpainting_mask_invert"
:
0
,
"prompt"
:
"example prompt"
,
"prompt"
:
"example prompt"
,
"styles"
:
[
"styles"
:
[],
""
],
"seed"
:
-
1
,
"seed"
:
-
1
,
"subseed"
:
-
1
,
"subseed"
:
-
1
,
"subseed_strength"
:
0
,
"subseed_strength"
:
0
,
...
@@ -45,6 +42,7 @@ class TestImg2ImgWorking(unittest.TestCase):
...
@@ -45,6 +42,7 @@ class TestImg2ImgWorking(unittest.TestCase):
"sampler_index"
:
"Euler a"
,
"sampler_index"
:
"Euler a"
,
"include_init_images"
:
False
"include_init_images"
:
False
}
}
def
test_img2img_simple_performed
(
self
):
def
test_img2img_simple_performed
(
self
):
self
.
assertEqual
(
requests
.
post
(
self
.
url_img2img
,
json
=
self
.
simple_img2img
)
.
status_code
,
200
)
self
.
assertEqual
(
requests
.
post
(
self
.
url_img2img
,
json
=
self
.
simple_img2img
)
.
status_code
,
200
)
...
@@ -52,8 +50,10 @@ class TestImg2ImgWorking(unittest.TestCase):
...
@@ -52,8 +50,10 @@ class TestImg2ImgWorking(unittest.TestCase):
self
.
simple_img2img
[
"mask"
]
=
encode_pil_to_base64
(
Image
.
open
(
r"test/test_files/mask_basic.png"
))
self
.
simple_img2img
[
"mask"
]
=
encode_pil_to_base64
(
Image
.
open
(
r"test/test_files/mask_basic.png"
))
self
.
assertEqual
(
requests
.
post
(
self
.
url_img2img
,
json
=
self
.
simple_img2img
)
.
status_code
,
200
)
self
.
assertEqual
(
requests
.
post
(
self
.
url_img2img
,
json
=
self
.
simple_img2img
)
.
status_code
,
200
)
class
TestImg2ImgCorrectness
(
unittest
.
TestCase
):
class
TestImg2ImgCorrectness
(
unittest
.
TestCase
):
pass
pass
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
unittest
.
main
()
unittest
.
main
()
test/server_poll.py
View file @
59dfe084
...
@@ -2,16 +2,18 @@ import unittest
...
@@ -2,16 +2,18 @@ import unittest
import
requests
import
requests
import
time
import
time
timeout_threshold
=
240
start_time
=
time
.
time
()
def
run_tests
():
while
time
.
time
()
-
start_time
<
timeout_threshold
:
timeout_threshold
=
240
try
:
start_time
=
time
.
time
()
requests
.
head
(
"http://localhost:7860/"
)
while
time
.
time
()
-
start_time
<
timeout_threshold
:
break
try
:
except
requests
.
exceptions
.
ConnectionError
:
requests
.
head
(
"http://localhost:7860/"
)
pass
break
if
time
.
time
()
-
start_time
<
timeout_threshold
:
except
requests
.
exceptions
.
ConnectionError
:
suite
=
unittest
.
TestLoader
()
.
discover
(
''
,
pattern
=
'*_test.py'
)
pass
result
=
unittest
.
TextTestRunner
(
verbosity
=
2
)
.
run
(
suite
)
if
time
.
time
()
-
start_time
<
timeout_threshold
:
else
:
suite
=
unittest
.
TestLoader
()
.
discover
(
''
,
pattern
=
'*_test.py'
)
print
(
"Launch unsuccessful"
)
result
=
unittest
.
TextTestRunner
(
verbosity
=
2
)
.
run
(
suite
)
else
:
print
(
"Launch unsuccessful"
)
test/txt2img_test.py
View file @
59dfe084
import
unittest
import
unittest
import
requests
import
requests
class
TestTxt2ImgWorking
(
unittest
.
TestCase
):
class
TestTxt2ImgWorking
(
unittest
.
TestCase
):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
url_txt2img
=
"http://localhost:7860/sdapi/v1/txt2img"
self
.
url_txt2img
=
"http://localhost:7860/sdapi/v1/txt2img"
...
@@ -10,9 +11,7 @@ class TestTxt2ImgWorking(unittest.TestCase):
...
@@ -10,9 +11,7 @@ class TestTxt2ImgWorking(unittest.TestCase):
"firstphase_width"
:
0
,
"firstphase_width"
:
0
,
"firstphase_height"
:
0
,
"firstphase_height"
:
0
,
"prompt"
:
"example prompt"
,
"prompt"
:
"example prompt"
,
"styles"
:
[
"styles"
:
[],
""
],
"seed"
:
-
1
,
"seed"
:
-
1
,
"subseed"
:
-
1
,
"subseed"
:
-
1
,
"subseed_strength"
:
0
,
"subseed_strength"
:
0
,
...
@@ -34,6 +33,7 @@ class TestTxt2ImgWorking(unittest.TestCase):
...
@@ -34,6 +33,7 @@ class TestTxt2ImgWorking(unittest.TestCase):
"s_noise"
:
1
,
"s_noise"
:
1
,
"sampler_index"
:
"Euler a"
"sampler_index"
:
"Euler a"
}
}
def
test_txt2img_simple_performed
(
self
):
def
test_txt2img_simple_performed
(
self
):
self
.
assertEqual
(
requests
.
post
(
self
.
url_txt2img
,
json
=
self
.
simple_txt2img
)
.
status_code
,
200
)
self
.
assertEqual
(
requests
.
post
(
self
.
url_txt2img
,
json
=
self
.
simple_txt2img
)
.
status_code
,
200
)
...
@@ -65,8 +65,10 @@ class TestTxt2ImgWorking(unittest.TestCase):
...
@@ -65,8 +65,10 @@ class TestTxt2ImgWorking(unittest.TestCase):
self
.
simple_txt2img
[
"n_iter"
]
=
2
self
.
simple_txt2img
[
"n_iter"
]
=
2
self
.
assertEqual
(
requests
.
post
(
self
.
url_txt2img
,
json
=
self
.
simple_txt2img
)
.
status_code
,
200
)
self
.
assertEqual
(
requests
.
post
(
self
.
url_txt2img
,
json
=
self
.
simple_txt2img
)
.
status_code
,
200
)
class
TestTxt2ImgCorrectness
(
unittest
.
TestCase
):
class
TestTxt2ImgCorrectness
(
unittest
.
TestCase
):
pass
pass
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
unittest
.
main
()
unittest
.
main
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment