• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""Tests to improve the the of tensorflow.
2
3Here we would like to include high level tests that stress tf.function and
4autograph in ways users have discovered.  Not everything here has to work,
5some things just need to have good error messages.  some things currently
6have bugs assigned to them but do not work and do not have sufficient error
7messages.
8"""
9
10load(
11    "//tensorflow:tensorflow.bzl",
12    "py_test",
13)
14
15package(
16    default_visibility = ["//tensorflow/tools/consistency_integration_test:__subpackages__"],
17    licenses = ["notice"],
18)
19
20licenses(["notice"])
21
22py_library(
23    name = "consistency_test_base",
24    srcs = ["consistency_test_base.py"],
25    srcs_version = "PY3",
26    deps = [
27        "//tensorflow:tensorflow_py",
28        "//third_party/py/numpy",
29        "@six_archive//:six",
30        "@termcolor_archive//:termcolor",
31    ],
32)
33
34py_test(
35    name = "consistency_test_base_tests",
36    srcs = ["consistency_test_base_tests.py"],
37    srcs_version = "PY3",
38    tags = [
39        "no_pip",
40    ],
41    deps = [
42        ":consistency_test_base",
43        "//tensorflow/python:client_testlib",
44    ],
45)
46
47py_test(
48    name = "function_io_tests",
49    srcs = ["function_io_tests.py"],
50    srcs_version = "PY3",
51    tags = [
52        "no_pip",
53    ],
54    deps = [
55        ":consistency_test_base",
56        "//tensorflow/python:client_testlib",
57        "@absl_py//absl/testing:parameterized",
58    ],
59)
60
61py_test(
62    name = "tensor_array_tests",
63    srcs = ["tensor_array_tests.py"],
64    srcs_version = "PY3",
65    tags = [
66        "no_pip",
67    ],
68    deps = [
69        ":consistency_test_base",
70        "//tensorflow/python:client_testlib",
71    ],
72)
73
74py_test(
75    name = "type_promotion_tests",
76    srcs = ["type_promotion_tests.py"],
77    srcs_version = "PY3",
78    tags = [
79        "no_pip",
80    ],
81    deps = [
82        ":consistency_test_base",
83        "//tensorflow/python:client_testlib",
84    ],
85)
86
87py_test(
88    name = "keras_integration_tests",
89    srcs = ["keras_integration_tests.py"],
90    srcs_version = "PY3",
91    tags = [
92        "no_pip",
93    ],
94    deps = [
95        ":consistency_test_base",
96        "//tensorflow/python:client_testlib",
97        "@absl_py//absl/testing:parameterized",
98    ],
99)
100