Home | History | Annotate | Download | only in kernel_tests 
      1 package(default_visibility = ["//tensorflow:internal"])
      2 
      3 licenses(["notice"])  # Apache 2.0
      4 
      5 exports_files(["LICENSE"])
      6 
      7 load("//tensorflow:tensorflow.bzl", "py_test", "tf_py_test")
      8 
      9 py_test(
     10     name = "batch_dataset_op_test",
     11     size = "small",
     12     srcs = ["batch_dataset_op_test.py"],
     13     srcs_version = "PY2AND3",
     14     tags = ["no_pip"],
     15     deps = [
     16         ":dataset_serialization_test",
     17         "//tensorflow/contrib/data/python/ops:dataset_ops",
     18         "//tensorflow/contrib/data/python/ops:transformation_ops",
     19         "//tensorflow/python:array_ops",
     20         "//tensorflow/python:client_testlib",
     21         "//tensorflow/python:constant_op",
     22         "//tensorflow/python:dtypes",
     23         "//tensorflow/python:errors",
     24         "//tensorflow/python:math_ops",
     25         "//tensorflow/python:sparse_tensor",
     26         "//tensorflow/python:string_ops",
     27         "//tensorflow/python:tensor_shape",
     28         "//tensorflow/python:util",
     29         "//third_party/py/numpy",
     30     ],
     31 )
     32 
     33 py_test(
     34     name = "bucketing_test",
     35     size = "small",
     36     srcs = ["bucketing_test.py"],
     37     srcs_version = "PY2AND3",
     38     deps = [
     39         ":dataset_serialization_test",
     40         "//tensorflow/contrib/data/python/ops:dataset_ops",
     41         "//tensorflow/contrib/data/python/ops:transformation_ops",
     42         "//tensorflow/python:array_ops",
     43         "//tensorflow/python:client_testlib",
     44         "//tensorflow/python:constant_op",
     45         "//tensorflow/python:dtypes",
     46         "//tensorflow/python:errors",
     47         "//tensorflow/python:framework_ops",
     48         "//tensorflow/python:math_ops",
     49         "//tensorflow/python:string_ops",
     50         "//tensorflow/python:tensor_shape",
     51         "//third_party/py/numpy",
     52     ],
     53 )
     54 
     55 py_test(
     56     name = "concatenate_dataset_op_test",
     57     size = "small",
     58     srcs = ["concatenate_dataset_op_test.py"],
     59     srcs_version = "PY2AND3",
     60     deps = [
     61         ":dataset_serialization_test",
     62         "//tensorflow/contrib/data/python/ops:dataset_ops",
     63         "//tensorflow/python:client_testlib",
     64         "//tensorflow/python:errors",
     65         "//tensorflow/python:tensor_shape",
     66         "//tensorflow/python/data/util:nest",
     67         "//third_party/py/numpy",
     68     ],
     69 )
     70 
     71 py_test(
     72     name = "dataset_constructor_op_test",
     73     size = "medium",
     74     srcs = ["dataset_constructor_op_test.py"],
     75     srcs_version = "PY2AND3",
     76     tags = [
     77         "manual",
     78         "nomac",  # b/62040583
     79     ],
     80     deps = [
     81         ":dataset_serialization_test",
     82         "//tensorflow/contrib/data/python/ops:dataset_ops",
     83         "//tensorflow/contrib/data/python/ops:transformation_ops",
     84         "//tensorflow/core:protos_all_py",
     85         "//tensorflow/python:array_ops",
     86         "//tensorflow/python:client_testlib",
     87         "//tensorflow/python:dtypes",
     88         "//tensorflow/python:errors",
     89         "//tensorflow/python:framework_ops",
     90         "//tensorflow/python:math_ops",
     91         "//tensorflow/python:resource_variable_ops",
     92         "//tensorflow/python:session",
     93         "//tensorflow/python:sparse_tensor",
     94         "//tensorflow/python:tensor_shape",
     95         "//tensorflow/python/data/util:nest",
     96         "//third_party/py/numpy",
     97     ],
     98 )
     99 
    100 py_library(
    101     name = "dataset_serialization_test",
    102     srcs = [
    103         "dataset_serialization_test_base.py",
    104     ],
    105     srcs_version = "PY2AND3",
    106     deps = [
    107         "//tensorflow/contrib/data/python/ops:iterator_ops",
    108         "//tensorflow/python:client_testlib",
    109         "//tensorflow/python:errors",
    110         "//tensorflow/python:framework_ops",
    111         "//tensorflow/python:lookup_ops",
    112         "//tensorflow/python:platform",
    113         "//tensorflow/python:sparse_tensor",
    114         "//tensorflow/python:training",
    115         "//tensorflow/python:util",
    116         "//tensorflow/python:variables",
    117         "//tensorflow/python/data/ops:iterator_ops",
    118         "//third_party/py/numpy",
    119     ],
    120 )
    121 
    122 py_test(
    123     name = "filter_dataset_op_test",
    124     size = "small",
    125     srcs = ["filter_dataset_op_test.py"],
    126     srcs_version = "PY2AND3",
    127     tags = ["no_pip"],
    128     deps = [
    129         ":dataset_serialization_test",
    130         "//tensorflow/contrib/data/python/ops:dataset_ops",
    131         "//tensorflow/python:array_ops",
    132         "//tensorflow/python:client_testlib",
    133         "//tensorflow/python:dtypes",
    134         "//tensorflow/python:errors",
    135         "//tensorflow/python:functional_ops",
    136         "//tensorflow/python:math_ops",
    137         "//third_party/py/numpy",
    138     ],
    139 )
    140 
    141 tf_py_test(
    142     name = "flat_map_dataset_op_test",
    143     size = "medium",
    144     srcs = ["flat_map_dataset_op_test.py"],
    145     additional_deps = [
    146         ":dataset_serialization_test",
    147         "//third_party/py/numpy",
    148         "//tensorflow/contrib/data/python/ops:dataset_ops",
    149         "//tensorflow/python:array_ops",
    150         "//tensorflow/python:client_testlib",
    151         "//tensorflow/python:constant_op",
    152         "//tensorflow/python:dtypes",
    153         "//tensorflow/python:errors",
    154         "//tensorflow/python:function",
    155         "//tensorflow/python:math_ops",
    156         "//tensorflow/python:random_ops",
    157         "//tensorflow/python:session",
    158         "//tensorflow/python:training",
    159         "//tensorflow/python:variable_scope",
    160     ],
    161     grpc_enabled = True,
    162     tags = ["no_pip"],
    163 )
    164 
    165 py_test(
    166     name = "interleave_dataset_op_test",
    167     size = "medium",
    168     srcs = ["interleave_dataset_op_test.py"],
    169     srcs_version = "PY2AND3",
    170     tags = [
    171         "no_oss",
    172         "no_pip",
    173     ],
    174     deps = [
    175         ":dataset_serialization_test",
    176         "//tensorflow/contrib/data/python/ops:dataset_ops",
    177         "//tensorflow/contrib/data/python/ops:transformation_ops",
    178         "//tensorflow/python:array_ops",
    179         "//tensorflow/python:client",
    180         "//tensorflow/python:client_testlib",
    181         "//tensorflow/python:dtypes",
    182         "//tensorflow/python:errors",
    183         "//tensorflow/python:math_ops",
    184         "//tensorflow/python:script_ops",
    185         "//tensorflow/python:sparse_ops",
    186         "//tensorflow/python:sparse_tensor",
    187         "//tensorflow/python:training",
    188         "//third_party/py/numpy",
    189     ],
    190 )
    191 
    192 tf_py_test(
    193     name = "get_single_element_test",
    194     size = "small",
    195     srcs = ["get_single_element_test.py"],
    196     additional_deps = [
    197         "//third_party/py/numpy",
    198         "//tensorflow/contrib/data/python/ops:dataset_ops",
    199         "//tensorflow/python:array_ops",
    200         "//tensorflow/python:client_testlib",
    201         "//tensorflow/python:constant_op",
    202         "//tensorflow/python:dtypes",
    203         "//tensorflow/python:errors",
    204         "//tensorflow/python:framework_test_lib",
    205     ],
    206 )
    207 
    208 py_test(
    209     name = "map_dataset_op_test",
    210     size = "medium",
    211     srcs = ["map_dataset_op_test.py"],
    212     srcs_version = "PY2AND3",
    213     tags = ["no_pip"],
    214     deps = [
    215         ":dataset_serialization_test",
    216         "//tensorflow/contrib/data/python/ops:dataset_ops",
    217         "//tensorflow/contrib/data/python/ops:transformation_ops",
    218         "//tensorflow/python:array_ops",
    219         "//tensorflow/python:client_testlib",
    220         "//tensorflow/python:constant_op",
    221         "//tensorflow/python:data_flow_ops",
    222         "//tensorflow/python:dtypes",
    223         "//tensorflow/python:errors",
    224         "//tensorflow/python:framework_ops",
    225         "//tensorflow/python:function",
    226         "//tensorflow/python:functional_ops",
    227         "//tensorflow/python:io_ops",
    228         "//tensorflow/python:lookup_ops",
    229         "//tensorflow/python:math_ops",
    230         "//tensorflow/python:random_ops",
    231         "//tensorflow/python:script_ops",
    232         "//tensorflow/python:sparse_ops",
    233         "//tensorflow/python:sparse_tensor",
    234         "//tensorflow/python:string_ops",
    235         "//tensorflow/python:util",
    236         "//tensorflow/python:variable_scope",
    237         "//tensorflow/python/data/ops:dataset_ops",
    238         "//third_party/py/numpy",
    239     ],
    240 )
    241 
    242 py_test(
    243     name = "prefetch_dataset_op_test",
    244     size = "small",
    245     srcs = ["prefetch_dataset_op_test.py"],
    246     srcs_version = "PY2AND3",
    247     tags = ["no_pip"],
    248     deps = [
    249         ":dataset_serialization_test",
    250         "//tensorflow/python:platform",
    251         "//tensorflow/python/data/ops:dataset_ops",
    252     ],
    253 )
    254 
    255 py_test(
    256     name = "range_dataset_op_test",
    257     size = "small",
    258     srcs = ["range_dataset_op_test.py"],
    259     srcs_version = "PY2AND3",
    260     deps = [
    261         ":dataset_serialization_test",
    262         "//tensorflow/contrib/data/python/ops:dataset_ops",
    263         "//tensorflow/contrib/data/python/ops:transformation_ops",
    264         "//tensorflow/python:array_ops",
    265         "//tensorflow/python:client_testlib",
    266         "//tensorflow/python:constant_op",
    267         "//tensorflow/python:dataset_ops_gen",
    268         "//tensorflow/python:dtypes",
    269         "//tensorflow/python:errors",
    270         "//tensorflow/python:framework_ops",
    271         "//tensorflow/python:io_ops",
    272         "//tensorflow/python:parsing_ops",
    273         "//tensorflow/python:tensor_shape",
    274         "//tensorflow/python:variables",
    275     ],
    276 )
    277 
    278 py_test(
    279     name = "reader_dataset_ops_test",
    280     size = "medium",
    281     srcs = ["reader_dataset_ops_test.py"],
    282     srcs_version = "PY2AND3",
    283     tags = ["no_pip"],
    284     deps = [
    285         ":dataset_serialization_test",
    286         "//tensorflow/contrib/data/python/ops:readers",
    287         "//tensorflow/core:protos_all_py",
    288         "//tensorflow/python:array_ops",
    289         "//tensorflow/python:client_testlib",
    290         "//tensorflow/python:constant_op",
    291         "//tensorflow/python:dtypes",
    292         "//tensorflow/python:errors",
    293         "//tensorflow/python:framework_ops",
    294         "//tensorflow/python:lib",
    295         "//tensorflow/python:parsing_ops",
    296         "//tensorflow/python:util",
    297         "//tensorflow/python/data/ops:iterator_ops",
    298     ],
    299 )
    300 
    301 py_test(
    302     name = "resample_test",
    303     size = "medium",
    304     srcs = ["resample_test.py"],
    305     shard_count = 2,
    306     srcs_version = "PY2AND3",
    307     tags = ["noasan"],
    308     deps = [
    309         "//tensorflow/contrib/data/python/ops:dataset_ops",
    310         "//tensorflow/contrib/data/python/ops:transformation_ops",
    311         "//tensorflow/python:client_testlib",
    312         "//tensorflow/python:errors",
    313         "//tensorflow/python:string_ops",
    314         "//tensorflow/python:util",
    315         "//third_party/py/numpy",
    316     ],
    317 )
    318 
    319 py_test(
    320     name = "scan_dataset_op_test",
    321     size = "small",
    322     srcs = ["scan_dataset_op_test.py"],
    323     srcs_version = "PY2AND3",
    324     tags = ["no_pip"],
    325     deps = [
    326         ":dataset_serialization_test",
    327         "//tensorflow/contrib/data/python/ops:transformation_ops",
    328         "//tensorflow/python:array_ops",
    329         "//tensorflow/python:client_testlib",
    330         "//tensorflow/python:constant_op",
    331         "//tensorflow/python:dtypes",
    332         "//tensorflow/python:errors",
    333         "//tensorflow/python/data/ops:dataset_ops",
    334         "//third_party/py/numpy",
    335     ],
    336 )
    337 
    338 py_test(
    339     name = "sequence_dataset_op_test",
    340     size = "medium",
    341     srcs = ["sequence_dataset_op_test.py"],
    342     srcs_version = "PY2AND3",
    343     tags = ["no_pip"],
    344     deps = [
    345         ":dataset_serialization_test",
    346         "//tensorflow/contrib/data/python/ops:dataset_ops",
    347         "//tensorflow/python:array_ops",
    348         "//tensorflow/python:client_testlib",
    349         "//tensorflow/python:dtypes",
    350         "//tensorflow/python:errors",
    351         "//third_party/py/numpy",
    352     ],
    353 )
    354 
    355 py_test(
    356     name = "serialization_integration_test",
    357     size = "small",
    358     srcs = ["serialization_integration_test.py"],
    359     srcs_version = "PY2AND3",
    360     tags = ["no_pip"],
    361     deps = [
    362         "//tensorflow/contrib/data/python/ops:iterator_ops",
    363         "//tensorflow/python:client_testlib",
    364         "//tensorflow/python:framework_ops",
    365         "//tensorflow/python:training",
    366         "//tensorflow/python/data/ops:dataset_ops",
    367     ],
    368 )
    369 
    370 py_test(
    371     name = "shuffle_dataset_op_test",
    372     size = "medium",
    373     srcs = ["shuffle_dataset_op_test.py"],
    374     srcs_version = "PY2AND3",
    375     tags = ["no_pip"],
    376     deps = [
    377         ":dataset_serialization_test",
    378         "//tensorflow/contrib/data/python/ops:dataset_ops",
    379         "//tensorflow/contrib/data/python/ops:shuffle_ops",
    380         "//tensorflow/python:array_ops",
    381         "//tensorflow/python:client_testlib",
    382         "//tensorflow/python:constant_op",
    383         "//tensorflow/python:dtypes",
    384         "//tensorflow/python:errors",
    385         "//tensorflow/python:framework_ops",
    386         "//tensorflow/python/data/ops:dataset_ops",
    387         "//tensorflow/python/data/ops:iterator_ops",
    388         "//third_party/py/numpy",
    389     ],
    390 )
    391 
    392 py_test(
    393     name = "sql_dataset_op_test",
    394     size = "small",
    395     srcs = ["sql_dataset_op_test.py"],
    396     srcs_version = "PY2AND3",
    397     deps = [
    398         "//tensorflow/contrib/data/python/ops:readers",
    399         "//tensorflow/python:array_ops",
    400         "//tensorflow/python:client_testlib",
    401         "//tensorflow/python:dtypes",
    402         "//tensorflow/python:errors",
    403         "@org_sqlite//:python",
    404     ],
    405 )
    406 
    407 py_test(
    408     name = "stats_dataset_ops_test",
    409     size = "medium",
    410     srcs = ["stats_dataset_ops_test.py"],
    411     srcs_version = "PY2AND3",
    412     tags = ["no_pip"],
    413     deps = [
    414         ":dataset_serialization_test",
    415         "//tensorflow/contrib/data/python/ops:dataset_ops",
    416         "//tensorflow/contrib/data/python/ops:transformation_ops",
    417         "//tensorflow/python:client_testlib",
    418         "//tensorflow/python:errors",
    419     ],
    420 )
    421 
    422 py_test(
    423     name = "unique_dataset_op_test",
    424     size = "small",
    425     srcs = ["unique_dataset_op_test.py"],
    426     srcs_version = "PY2AND3",
    427     tags = ["no_pip"],
    428     deps = [
    429         ":dataset_serialization_test",
    430         "//tensorflow/contrib/data/python/ops:dataset_ops",
    431         "//tensorflow/contrib/data/python/ops:transformation_ops",
    432         "//tensorflow/contrib/stateless",
    433         "//tensorflow/python:array_ops",
    434         "//tensorflow/python:client_testlib",
    435         "//tensorflow/python:dtypes",
    436         "//tensorflow/python:errors",
    437         "//third_party/py/numpy",
    438     ],
    439 )
    440 
    441 py_test(
    442     name = "zip_dataset_op_test",
    443     size = "small",
    444     srcs = ["zip_dataset_op_test.py"],
    445     srcs_version = "PY2AND3",
    446     tags = ["no_pip"],
    447     deps = [
    448         ":dataset_serialization_test",
    449         "//tensorflow/contrib/data/python/ops:dataset_ops",
    450         "//tensorflow/python:array_ops",
    451         "//tensorflow/python:client_testlib",
    452         "//tensorflow/python:dtypes",
    453         "//tensorflow/python:errors",
    454         "//third_party/py/numpy",
    455     ],
    456 )
    457 
    458 py_test(
    459     name = "prefetching_ops_test",
    460     size = "small",
    461     srcs = ["prefetching_ops_test.py"],
    462     srcs_version = "PY2AND3",
    463     tags = [
    464         "manual",
    465         "no_oss",  # b/68785503
    466     ],
    467     deps = [
    468         "//tensorflow/contrib/data/python/ops:prefetching_ops",
    469         "//tensorflow/core:protos_all_py",
    470         "//tensorflow/python:client_testlib",
    471         "//tensorflow/python:constant_op",
    472         "//tensorflow/python:dtypes",
    473         "//tensorflow/python:framework_ops",
    474         "//tensorflow/python:framework_test_lib",
    475         "//tensorflow/python:function",
    476         "//tensorflow/python:resource_variable_ops",
    477         "//tensorflow/python/data/ops:dataset_ops",
    478         "//tensorflow/python/data/ops:iterator_ops",
    479     ],
    480 )
    481 
    482 filegroup(
    483     name = "all_files",
    484     srcs = glob(
    485         include = [
    486             "**/*",
    487         ],
    488         exclude = [
    489             "**/METADATA",
    490             "**/OWNERS",
    491         ],
    492     ),
    493     visibility = ["//tensorflow:__subpackages__"],
    494 )
    495