derrickaw commented on code in PR #35435: URL: https://github.com/apache/beam/pull/35435#discussion_r2167667124
########## sdks/python/apache_beam/yaml/integration_tests.py: ########## @@ -39,22 +45,47 @@ from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs from testcontainers.google import PubSubContainer -from testcontainers.kafka import KafkaContainer +# from testcontainers.kafka import KafkaContainer from testcontainers.mssql import SqlServerContainer from testcontainers.mysql import MySqlContainer from testcontainers.postgres import PostgresContainer +import pytest + + import apache_beam as beam from apache_beam.io import filesystems from apache_beam.io.gcp.bigquery_tools import BigQueryWrapper from apache_beam.io.gcp.internal.clients import bigquery +from apache_beam.io.gcp import bigtableio + from apache_beam.io.gcp.spanner_wrapper import SpannerWrapper from apache_beam.options.pipeline_options import PipelineOptions from apache_beam.utils import python_callable from apache_beam.yaml import yaml_provider from apache_beam.yaml import yaml_transform from apache_beam.yaml.conftest import yaml_test_files_dir +from apache_beam.testing.test_pipeline import TestPipeline +from apache_beam.testing.util import assert_that +from apache_beam.testing.util import equal_to + +from google.cloud.bigtable import client +from google.cloud.bigtable_admin_v2.types import instance + +_LOGGER = logging.getLogger(__name__) + +# Protect against environments where bigtable library is not available. Review Comment: Have you seen this in your environment? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: github-unsubscr...@beam.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
