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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
'''
PyMOLWin.exe launch tests
'''
import os
import time
import subprocess
import sys
import unittest
from pymol import cmd, CmdException, testing, stored
@unittest.skipIf(not sys.platform.startswith('win'), 'windows only')
class TestPyMOLWin(testing.PyMOLTestCase):
def _get_pymolwin(self):
pymolwin = os.path.join(os.path.dirname(sys.executable), 'PyMOLWin.exe')
if os.path.exists(pymolwin):
return pymolwin
self.skipTest('missing ' + pymolwin)
@testing.requires_version('2.0')
def test_wchar(self):
'''
Wide-char file names
'''
pymolwin = self._get_pymolwin()
with testing.mkdtemp() as base:
dirname = u'Japanese \u65e5\u672c\u8a9e'
dirname = os.path.join(base, dirname)
os.mkdir(dirname)
out_filename = base + '\\out.pdb'
bat_filename = base + '\\in.bat'
pml_filename = dirname + u'\\in.pml'
in_filename = dirname + u'\\in.pdb'
command = u'"%s" -kcq "%s" "%s"' % (pymolwin, in_filename, pml_filename)
cmd.fragment('gly')
cmd.save(in_filename)
cmd.delete('*')
with open(pml_filename, 'w') as handle:
handle.write('save %s\n' % out_filename)
with open(bat_filename, 'wb') as handle:
handle.write(b'chcp 65001\r\n') # UTF-8 code page
handle.write(b'set PYTHONIOENCODING=UTF-8\r\n') # avoid LookupError
handle.write(command.encode('utf-8') + b'\r\n')
handle.write(b'chcp 437\r\n')
subprocess.call([bat_filename])
# wait for spawned process
for _ in range(10):
if os.path.exists(out_filename):
break
time.sleep(0.5)
cmd.load(out_filename)
self.assertEqual(7, cmd.count_atoms())
def test_unc(self):
'''
UNC-Paths (PYMOL-2954)
'''
pymolwin = self._get_pymolwin()
with testing.mkdtemp() as base:
if base[1] != ':':
self.skipTest('no drive letter: ' + base)
return
if base.startswith('\\\\'):
uncpath = base
else:
uncpath = '\\\\localhost\\%s$%s' % (base[0], base[2:])
pml_filename = uncpath + '\\in.pml'
out_filename = uncpath + '\\out.pdb'
with open(pml_filename, 'w') as handle:
handle.write('fragment gly\n')
handle.write('save %s\n' % out_filename)
subprocess.call([pymolwin, '+2', '-kcq', pml_filename])
# wait for spawned process
time.sleep(0.5)
cmd.load(out_filename)
self.assertEqual(7, cmd.count_atoms())
|
