Commit | Line | Data |
---|---|---|
0f9675ba FC |
1 | from pathlib import Path |
2 | import sys | |
3 | if '' in sys.path: sys.path.remove('') | |
4 | sys.path.append(str(Path(__file__).parent.parent.parent)) | |
0f9675ba FC |
5 | from unittest import TestCase |
6 | from setuptools.dist import Distribution | |
7 | from setup import AbsCmd | |
53ddf3c3 | 8 | from ya2.build.build import exec_cmd |
0f9675ba FC |
9 | |
10 | ||
11 | class SetupTests(TestCase): | |
12 | ||
13 | def setUp(self): | |
14 | self._prev_argv = sys.argv | |
15 | ||
16 | def tearDown(self): | |
17 | sys.argv = self._prev_argv | |
18 | ||
19 | def test_cores(self): | |
20 | cmd = AbsCmd(Distribution()) | |
addec9c9 FC |
21 | ncores = int(exec_cmd('grep -c ^processor /proc/cpuinfo')) |
22 | self.assertEqual(cmd.cores, ncores) | |
0f9675ba FC |
23 | sys.argv += ['--cores=2'] |
24 | cmd = AbsCmd(Distribution()) | |
25 | self.assertEqual(cmd.cores, 2) |