github.phpd.cn/thought-machine/please@v12.2.0+incompatible/test/python_rules/strip_source_test.py (about) 1 """Tests on source stripping.""" 2 3 import unittest 4 import zipfile 5 6 7 class StripSourceTest(unittest.TestCase): 8 9 def test_file_is_a_pyc(self): 10 """Test that the stripped module is a .pyc.""" 11 from test.python_rules import strip_source 12 self.assertTrue(strip_source.__file__.endswith('.pyc')) 13 14 def test_this_file_doesnt_exist(self): 15 """Test this file doesn't exist in the pex.""" 16 import __main__ as pex_main 17 with zipfile.ZipFile(pex_main.PEX, 'r') as zf: 18 with self.assertRaises(KeyError): 19 zf.getinfo('test/python_rules/strip_source.py')