@@ -33,8 +33,14 @@ def previous_job_run_id() -> int:
33
33
@pytest .mark .parametrize (
34
34
"deployed_connection_id" ,
35
35
[
36
- pytest .param ("c7b4d838-a612-495a-9d91-a14e477add51" , id = "Faker->Snowflake" ),
37
- pytest .param ("0e1d6b32-b8e3-4b68-91a3-3a314599c782" , id = "Faker->BigQuery" ),
36
+ pytest .param (
37
+ "c7b4d838-a612-495a-9d91-a14e477add51" , # https://cloud.airbyte.com/workspaces/a0cc325a-d358-4df4-bdd4-c09d753b6afb/connections/c7b4d838-a612-495a-9d91-a14e477add51/status
38
+ id = "Faker->Snowflake" ,
39
+ ),
40
+ pytest .param (
41
+ "0e1d6b32-b8e3-4b68-91a3-3a314599c782" , # https://cloud.airbyte.com/workspaces/a0cc325a-d358-4df4-bdd4-c09d753b6afb/connections/0e1d6b32-b8e3-4b68-91a3-3a314599c782/status
42
+ id = "Faker->BigQuery" ,
43
+ ),
38
44
pytest .param (
39
45
"" , id = "Faker->Postgres" , marks = pytest .mark .skip (reason = "Not yet supported" )
40
46
),
@@ -76,7 +82,11 @@ def test_read_from_deployed_connection(
76
82
77
83
pandas_df = pd .DataFrame (data_as_list )
78
84
79
- assert pandas_df .shape == (100 , 20 )
85
+ assert pandas_df .shape [0 ] == 100
86
+ assert pandas_df .shape [1 ] in { # Column count diff depending on when it was created
87
+ 20 ,
88
+ 21 ,
89
+ }
80
90
81
91
# Check that no values are null
82
92
for col in pandas_df .columns :
@@ -87,12 +97,12 @@ def test_read_from_deployed_connection(
87
97
"deployed_connection_id, cache_type" ,
88
98
[
89
99
pytest .param (
90
- "c7b4d838-a612-495a-9d91-a14e477add51" ,
100
+ "c7b4d838-a612-495a-9d91-a14e477add51" , # https://cloud.airbyte.com/workspaces/a0cc325a-d358-4df4-bdd4-c09d753b6afb/connections/c7b4d838-a612-495a-9d91-a14e477add51/status
91
101
SnowflakeCache ,
92
102
id = "Faker->Snowflake" ,
93
103
),
94
104
pytest .param (
95
- "0e1d6b32-b8e3-4b68-91a3-3a314599c782" ,
105
+ "0e1d6b32-b8e3-4b68-91a3-3a314599c782" , # https://cloud.airbyte.com/workspaces/a0cc325a-d358-4df4-bdd4-c09d753b6afb/connections/0e1d6b32-b8e3-4b68-91a3-3a314599c782/status
96
106
BigQueryCache ,
97
107
id = "Faker->BigQuery" ,
98
108
),
0 commit comments