-
Notifications
You must be signed in to change notification settings - Fork 78
/
test_smart2.py
48 lines (36 loc) · 1.38 KB
/
test_smart2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from easyprocess import EasyProcess
from nose.tools import eq_
from pyvirtualdisplay.smartdisplay import SmartDisplay, DisplayTimeoutError
from unittest import TestCase
from path import path
import pyscreenshot
class Test(TestCase):
def check_double(self, backend1, backend2=None):
if not backend2:
backend2 = backend1
with SmartDisplay(visible=0, bgcolor='black') as disp:
disp.pyscreenshot_backend = backend1
with EasyProcess('xmessage hello1'):
img = disp.waitgrab()
eq_(img is not None, True)
with SmartDisplay(visible=0, bgcolor='black') as disp:
disp.pyscreenshot_backend = backend2
with EasyProcess('xmessage hello2'):
img = disp.waitgrab()
eq_(img is not None, True)
def test_double_wx(self):
self.check_double('wx')
def test_double_pygtk(self):
self.check_double('pygtk')
def test_double_pyqt(self):
self.check_double('pyqt')
def test_double_imagemagick(self):
self.check_double('imagemagick')
def test_double_scrot(self):
self.check_double('scrot')
def test_double_wx_pygtk(self):
self.check_double('wx', 'pygtk')
def test_double_wx_pyqt(self):
self.check_double('wx', 'pyqt')
def test_double_pygtk_pyqt(self):
self.check_double('pygtk', 'pyqt')