added cascading flows test

Author: gisilvs

tensorflow_probability/python/experimental/vi/BUILD

@@ -74,7 +74,6 @@ py_library(
     srcs_version = "PY3",
     deps = [
         # tensorflow dep,
-
         "//tensorflow_probability/python/bijectors:build_highway_flow_layer",
         "//tensorflow_probability/python/bijectors:chain",
         "//tensorflow_probability/python/bijectors:reshape",
@@ -143,6 +142,22 @@ py_test(
     ],
 )
 
+py_test(
+    name = "cascading_flows_test",
+    size = "large",
+    srcs = ["cascading_flows_test.py"],
+    python_version = "PY3",
+    shard_count = 4,
+    srcs_version = "PY3",
+    deps = [
+        # absl/testing:parameterized dep,
+        # numpy dep,
+        # tensorflow dep,
+        "//tensorflow_probability",
+        "//tensorflow_probability/python/internal:test_util",
+    ],
+)
+
 py_test(
     name = "surrogate_posteriors_test",
     size = "large",