@@ -48,7 +48,7 @@ def test_fetch_sample_data_with_query(self):
4848 assert result == self .mock_response
4949
5050 def test_fetch_sample_data_with_languages_filter (self ):
51- filters = {LANGUAGES_FILTER . name : ["python" , "javascript" ]}
51+ filters = {LANGUAGES_FILTER [ " name" ] : ["python" , "javascript" ]}
5252
5353 result = fetch_sample_data (client = self .mock_client , query = "test query" , filters = filters , logger = self .mock_logger )
5454
@@ -57,41 +57,44 @@ def test_fetch_sample_data_with_languages_filter(self):
5757 assert result == self .mock_response
5858
5959 def test_fetch_sample_data_with_templates_filter (self ):
60- filters = {TEMPLATES_FILTER . name : True }
60+ filters = {TEMPLATES_FILTER [ " name" ] : True }
6161
6262 result = fetch_sample_data (client = self .mock_client , query = "test query" , filters = filters , logger = self .mock_logger )
6363
6464 self .mock_client .api_call .assert_called_once_with (
65- API_METHOD , params = {"query" : "test query" , "filters" : {"type" : TEMPLATES_FILTER . name }}
65+ API_METHOD , params = {"query" : "test query" , "filters" : {"type" : TEMPLATES_FILTER [ " name" ] }}
6666 )
6767
6868 assert result == self .mock_response
6969
7070 def test_fetch_sample_data_with_samples_filter (self ):
71- filters = {SAMPLES_FILTER . name : True }
71+ filters = {SAMPLES_FILTER [ " name" ] : True }
7272
7373 result = fetch_sample_data (client = self .mock_client , query = "test query" , filters = filters , logger = self .mock_logger )
7474
7575 self .mock_client .api_call .assert_called_once_with (
76- API_METHOD , params = {"query" : "test query" , "filters" : {"type" : SAMPLES_FILTER . name }}
76+ API_METHOD , params = {"query" : "test query" , "filters" : {"type" : SAMPLES_FILTER [ " name" ] }}
7777 )
7878
7979 assert result == self .mock_response
8080
8181 def test_fetch_sample_data_with_combined_filters (self ):
82- filters = {LANGUAGES_FILTER . name : ["python" ], TEMPLATES_FILTER . name : True }
82+ filters = {LANGUAGES_FILTER [ " name" ] : ["python" ], TEMPLATES_FILTER [ " name" ] : True }
8383
8484 result = fetch_sample_data (client = self .mock_client , query = "test query" , filters = filters , logger = self .mock_logger )
8585
8686 self .mock_client .api_call .assert_called_once_with (
8787 API_METHOD ,
88- params = {"query" : "test query" , "filters" : {LANGUAGES_FILTER .name : ["python" ], "type" : TEMPLATES_FILTER .name }},
88+ params = {
89+ "query" : "test query" ,
90+ "filters" : {LANGUAGES_FILTER ["name" ]: ["python" ], "type" : TEMPLATES_FILTER ["name" ]},
91+ },
8992 )
9093
9194 assert result == self .mock_response
9295
9396 def test_fetch_sample_data_with_both_template_and_sample (self ):
94- filters = {TEMPLATES_FILTER . name : True , SAMPLES_FILTER . name : True }
97+ filters = {TEMPLATES_FILTER [ " name" ] : True , SAMPLES_FILTER [ " name" ] : True }
9598
9699 result = fetch_sample_data (client = self .mock_client , query = "test query" , filters = filters , logger = self .mock_logger )
97100
0 commit comments