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
|
# Owner(s): ["oncall: mobile"]
import os
import tempfile
from pathlib import Path
from torch.jit.generate_bytecode import generate_upgraders_bytecode
from torch.testing._internal.common_utils import run_tests, TestCase
from torchgen.operator_versions.gen_mobile_upgraders import sort_upgrader, write_cpp
pytorch_caffe2_dir = Path(__file__).resolve().parents[2]
class TestLiteScriptModule(TestCase):
def test_generate_bytecode(self):
upgrader_list = generate_upgraders_bytecode()
sorted_upgrader_list = sort_upgrader(upgrader_list)
upgrader_mobile_cpp_path = (
pytorch_caffe2_dir
/ "torch"
/ "csrc"
/ "jit"
/ "mobile"
/ "upgrader_mobile.cpp"
)
with tempfile.TemporaryDirectory() as tmpdirname:
write_cpp(tmpdirname, sorted_upgrader_list)
with open(os.path.join(tmpdirname, "upgrader_mobile.cpp")) as file_name:
actual_output = [line.strip() for line in file_name if line]
with open(str(upgrader_mobile_cpp_path)) as file_name:
expect_output = [line.strip() for line in file_name if line]
actual_output_filtered = list(
filter(lambda token: len(token) != 0, actual_output)
)
expect_output_filtered = list(
filter(lambda token: len(token) != 0, expect_output)
)
self.assertEqual(actual_output_filtered, expect_output_filtered)
if __name__ == "__main__":
run_tests()
|
