@@ -24,7 +24,7 @@ def test_get_schedule_gtfs_dataset_key(self, project: str, dataset: str):
2424 assert result .feed_key .values [0 ] == "10f6bb537140b7e52c8f313f3d611f71"
2525
2626 @pytest .mark .vcr
27- def test_filter_dim_gtfs_dataset (self , project : str , dataset : str ):
27+ def test_filter_dim_gtfs_datasets (self , project : str , dataset : str ):
2828 result = filter_dim_gtfs_datasets (project = project , dataset = dataset )
2929
3030 assert len (result ) == 3
@@ -49,7 +49,7 @@ def test_filter_dim_gtfs_dataset(self, project: str, dataset: str):
4949 'base64_url' : 'aHR0cDovL2RhdGEucGVha3RyYW5zaXQuY29tL2d0ZnNydC8xL1RyaXBVcGRhdGUucGI=' }])
5050
5151 @pytest .mark .vcr
52- def test_filter_dim_gtfs_dataset_keep_cols_subset (self , project : str , dataset : str ):
52+ def test_filter_dim_gtfs_datasets_keep_cols_subset (self , project : str , dataset : str ):
5353 result = filter_dim_gtfs_datasets (keep_cols = ["key" , "name" ], project = project , dataset = dataset )
5454
5555 assert len (result ) == 3
@@ -62,7 +62,7 @@ def test_filter_dim_gtfs_dataset_keep_cols_subset(self, project: str, dataset: s
6262 'gtfs_dataset_name' : 'SLO Trip Updates' }])
6363
6464 @pytest .mark .vcr
65- def test_filter_dim_gtfs_dataset_custom_filtering (self , project : str , dataset : str ):
65+ def test_filter_dim_gtfs_datasets_custom_filtering (self , project : str , dataset : str ):
6666 result = filter_dim_gtfs_datasets (custom_filtering = {"type" : ["trip_updates" ]}, project = project , dataset = dataset )
6767
6868 assert len (result ) == 1
@@ -74,7 +74,7 @@ def test_filter_dim_gtfs_dataset_custom_filtering(self, project: str, dataset: s
7474 'uri' : 'http://data.peaktransit.com/gtfsrt/1/TripUpdate.pb' ,
7575 'base64_url' : 'aHR0cDovL2RhdGEucGVha3RyYW5zaXQuY29tL2d0ZnNydC8xL1RyaXBVcGRhdGUucGI=' }])
7676
77- def test_filter_dim_gtfs_dataset_keep_cols_key_missing (self , project : str , dataset : str ):
77+ def test_filter_dim_gtfs_datasets_keep_cols_key_missing (self , project : str , dataset : str ):
7878 with pytest .raises (KeyError , match = "Include key in keep_cols list" ):
7979 filter_dim_gtfs_datasets (keep_cols = ["name" , "type" , "regional_feed_type" , "uri" , "base64_url" ], project = project , dataset = dataset )
8080
0 commit comments