From 5707b3d4f4599e0884739fa6a6fb40e189fded27 Mon Sep 17 00:00:00 2001 From: Yogesh Tewari Date: Fri, 30 Jul 2021 15:05:49 -0400 Subject: [PATCH] feat: Allow user to specify a format for stdout, linting fix (#242) --- tests/system/data_sources/test_bigquery.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/system/data_sources/test_bigquery.py b/tests/system/data_sources/test_bigquery.py index b90c5a52b..21d9263d5 100644 --- a/tests/system/data_sources/test_bigquery.py +++ b/tests/system/data_sources/test_bigquery.py @@ -16,8 +16,8 @@ import pytest -from data_validation import cli_tools, consts, data_validation from data_validation import __main__ as main +from data_validation import cli_tools, consts, data_validation BQ_CONN = {"source_type": "BigQuery", "project_id": os.environ["PROJECT_ID"]} CONFIG_COUNT_VALID = { @@ -202,8 +202,7 @@ def test_count_validator(): assert float(max_birth_year_value) > 0 assert float(min_birth_year_value) > 0 assert ( - df["source_agg_value"].astype(float).sum() - == df["target_agg_value"].astype(float).sum() + df["source_agg_value"].astype(float).sum() == df["target_agg_value"].astype(float).sum() ) @@ -282,7 +281,7 @@ def _remove_bq_conn(): def test_unsupported_result_format(): - with pytest.raises(ValueError) as exp: + with pytest.raises(ValueError): validator = data_validation.DataValidation(CONFIG_GROUPED_COUNT_VALID, format="foobar", verbose=True) df = validator.execute() rows = list(df[df["validation_name"] == "count"].iterrows())