# This Source Code Form is subject to the terms of the Mozilla Public
# License, v.2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.

load("@integration_py_deps//:requirements.bzl", "requirement")

py_test(
    name = "integration",
    srcs = ["integration.py"],
    data = ["//embeddingstore:main"],
    main = "integration.py",
    deps = [
        requirement("pytest"),
        "//sdk/python:client",
    ],
)

py_test(
    name = "benchmark",
    size = "large",
    srcs = ["benchmark.py"],
    data = [
        ":glove_embeddings",
        "//embeddingstore:main",
    ],
    main = "benchmark.py",
    deps = [
        "//sdk/python:client",
    ],
)

filegroup(
    name = "glove_embeddings",
    srcs = glob([
        "glove*",
    ]),
)

filegroup(
    name = "all_source_files",
    srcs = glob([
        "BUILD",
        "*.py",
        "requirements.txt",
    ]),
    visibility = ["//visibility:public"],
)
