Skip to content

Commit f4472a2

Browse files
committed
fix: integ tests
1 parent ca2077e commit f4472a2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/integ/test_feature_store.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1089,9 +1089,9 @@ def test_get_feature_group_with_role_region(
10891089
athena_bucket=f"{offline_store_s3_uri}/query",
10901090
)
10911091

1092+
assert not dataset.empty
1093+
assert isinstance(dataset, DataFrame)
10921094
assert output["FeatureGroupArn"].endswith(f"feature-group/{feature_group_name}")
1093-
assert not dataset.empty
1094-
assert isinstance(dataset, DataFrame)
10951095

10961096

10971097
def test_get_feature_group_with_session(
@@ -1127,9 +1127,9 @@ def test_get_feature_group_with_session(
11271127
low_memory=False,
11281128
) # Using kwargs to pass a parameter to pandas.read_csv
11291129

1130+
assert not dataset.empty
1131+
assert isinstance(dataset, DataFrame)
11301132
assert output["FeatureGroupArn"].endswith(f"feature-group/{feature_group_name}")
1131-
assert not dataset.empty
1132-
assert isinstance(dataset, DataFrame)
11331133

11341134

11351135
@contextmanager

0 commit comments

Comments
 (0)