• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow:tensorflow.bzl", "py_test")
2
3package(
4    licenses = ["notice"],
5)
6
7filegroup(
8    name = "all_files",
9    srcs = glob(
10        ["**/*"],
11        exclude = [
12            "**/METADATA",
13            "**/OWNERS",
14        ],
15    ),
16    visibility = ["//visibility:private"],
17)
18
19py_library(
20    name = "converters",
21    srcs = [
22        "asserts.py",
23        "break_statements.py",
24        "call_trees.py",
25        "conditional_expressions.py",
26        "continue_statements.py",
27        "control_flow.py",
28        "control_flow_deprecated_py2.py",
29        "directives.py",
30        "functions.py",
31        "list_comprehensions.py",
32        "lists.py",
33        "logical_expressions.py",
34        "return_statements.py",
35        "slices.py",
36        "variables.py",
37    ],
38    srcs_version = "PY3",
39    visibility = ["//tensorflow:__subpackages__"],
40    deps = [
41        "//tensorflow/python:util",
42        "//tensorflow/python/autograph/core",
43        "//tensorflow/python/autograph/lang",
44        "//tensorflow/python/autograph/pyct",
45        "//tensorflow/python/autograph/pyct/static_analysis",
46        "//tensorflow/python/autograph/utils",
47        "@gast_archive//:gast",
48    ],
49)
50
51py_test(
52    name = "asserts_test",
53    srcs = ["asserts_test.py"],
54    python_version = "PY3",
55    srcs_version = "PY3",
56    deps = [
57        ":converters",
58        "//tensorflow/python:client_testlib",
59        "//tensorflow/python/autograph/core:test_lib",
60    ],
61)
62
63py_test(
64    name = "break_statements_test",
65    srcs = ["break_statements_test.py"],
66    python_version = "PY3",
67    srcs_version = "PY3",
68    deps = [
69        ":converters",
70        "//tensorflow/python:client_testlib",
71        "//tensorflow/python/autograph/core:test_lib",
72    ],
73)
74
75py_test(
76    name = "call_trees_test",
77    srcs = ["call_trees_test.py"],
78    python_version = "PY3",
79    srcs_version = "PY3",
80    deps = [
81        ":converters",
82        "//tensorflow/python:client_testlib",
83        "//tensorflow/python/autograph/core:test_lib",
84    ],
85)
86
87py_test(
88    name = "conditional_expressions_test",
89    srcs = ["conditional_expressions_test.py"],
90    python_version = "PY3",
91    srcs_version = "PY3",
92    deps = [
93        ":converters",
94        "//tensorflow/python:client_testlib",
95        "//tensorflow/python/autograph/core:test_lib",
96    ],
97)
98
99py_test(
100    name = "continue_statements_test",
101    srcs = ["continue_statements_test.py"],
102    python_version = "PY3",
103    srcs_version = "PY3",
104    deps = [
105        ":converters",
106        "//tensorflow/python:client_testlib",
107        "//tensorflow/python/autograph/core:test_lib",
108    ],
109)
110
111py_test(
112    name = "control_flow_test",
113    srcs = ["control_flow_test.py"],
114    python_version = "PY3",
115    srcs_version = "PY3",
116    deps = [
117        ":converters",
118        "//tensorflow/python:client_testlib",
119        "//tensorflow/python/autograph/core:test_lib",
120        # TOODO(b/145618471): Remove this transitive dependency.
121        "//tensorflow/python/distribute:input_lib",
122    ],
123)
124
125py_test(
126    name = "directives_test",
127    srcs = ["directives_test.py"],
128    python_version = "PY3",
129    srcs_version = "PY3",
130    deps = [
131        ":converters",
132        "//tensorflow/python:client_testlib",
133        "//tensorflow/python/autograph/core:test_lib",
134        "//tensorflow/python/autograph/lang",
135    ],
136)
137
138py_test(
139    name = "functions_test",
140    srcs = ["functions_test.py"],
141    python_version = "PY3",
142    deps = [
143        ":converters",
144        "//tensorflow/python:client_testlib",
145        "//tensorflow/python/autograph/core:test_lib",
146        "//tensorflow/python/autograph/pyct",
147    ],
148)
149
150py_test(
151    name = "list_comprehensions_test",
152    srcs = ["list_comprehensions_test.py"],
153    python_version = "PY3",
154    srcs_version = "PY3",
155    deps = [
156        ":converters",
157        "//tensorflow/python:client_testlib",
158        "//tensorflow/python/autograph/core:test_lib",
159    ],
160)
161
162py_test(
163    name = "lists_test",
164    srcs = ["lists_test.py"],
165    python_version = "PY3",
166    srcs_version = "PY3",
167    deps = [
168        ":converters",
169        "//tensorflow/python:client_testlib",
170        "//tensorflow/python/autograph/core:test_lib",
171    ],
172)
173
174py_test(
175    name = "logical_expressions_test",
176    srcs = ["logical_expressions_test.py"],
177    python_version = "PY3",
178    srcs_version = "PY3",
179    tags = ["notsan"],  # b/163218460
180    deps = [
181        ":converters",
182        "//tensorflow/python:client_testlib",
183        "//tensorflow/python/autograph/core:test_lib",
184    ],
185)
186
187py_test(
188    name = "return_statements_test",
189    srcs = ["return_statements_test.py"],
190    python_version = "PY3",
191    srcs_version = "PY3",
192    deps = [
193        ":converters",
194        "//tensorflow/python:client_testlib",
195        "//tensorflow/python/autograph/core:test_lib",
196        "//tensorflow/python/autograph/pyct",
197    ],
198)
199
200py_test(
201    name = "slices_test",
202    srcs = ["slices_test.py"],
203    python_version = "PY3",
204    srcs_version = "PY3",
205    deps = [
206        ":converters",
207        "//tensorflow/python:client_testlib",
208        "//tensorflow/python/autograph/core:test_lib",
209        "//tensorflow/python/autograph/pyct",
210    ],
211)
212
213py_test(
214    name = "variables_test",
215    srcs = ["variables_test.py"],
216    python_version = "PY3",
217    srcs_version = "PY3",
218    deps = [
219        ":converters",
220        "//tensorflow/python:client_testlib",
221        "//tensorflow/python/autograph/core:test_lib",
222        "//tensorflow/python/autograph/pyct",
223    ],
224)
225