20
20
override_application_settings ,
21
21
reset_core_stats_engine ,
22
22
)
23
+ from testing_support .fixtures import override_application_settings
23
24
from testing_support .validators .validate_ml_event_count import validate_ml_event_count
24
25
from testing_support .validators .validate_ml_events import validate_ml_events
25
26
@@ -284,8 +285,15 @@ def _test():
284
285
]
285
286
286
287
288
+ disabled_inference_value_settings = {
289
+ "machine_learning.enabled" : True ,
290
+ "machine_learning.inference_events_value.enabled" : False ,
291
+ "ml_insights_events.enabled" : True
292
+ }
293
+
294
+
295
+ @override_application_settings (disabled_inference_value_settings )
287
296
@reset_core_stats_engine ()
288
- @override_application_settings ({"machine_learning.inference_events_value.enabled" : False })
289
297
def test_does_not_include_value_when_inference_event_value_enabled_is_false ():
290
298
@validate_ml_events (numpy_str_recorded_custom_events_no_value )
291
299
@validate_ml_event_count (count = 1 )
@@ -306,8 +314,15 @@ def _test():
306
314
_test ()
307
315
308
316
317
+ disabled_ml_insights_settings = {
318
+ "machine_learning.enabled" : True ,
319
+ "machine_learning.inference_events_value.enabled" : True ,
320
+ "ml_insights_events.enabled" : False
321
+ }
322
+
323
+
324
+ @override_application_settings (disabled_ml_insights_settings )
309
325
@reset_core_stats_engine ()
310
- @override_application_settings ({"ml_insights_events.enabled" : False })
311
326
def test_does_not_include_events_when_ml_insights_events_enabled_is_false ():
312
327
"""
313
328
Verifies that all ml events can be disabled by setting
@@ -332,8 +347,15 @@ def _test():
332
347
_test ()
333
348
334
349
350
+ disabled_ml_settings = {
351
+ "machine_learning.enabled" : False ,
352
+ "machine_learning.inference_events_value.enabled" : True ,
353
+ "ml_insights_events.enabled" : True
354
+ }
355
+
356
+
357
+ @override_application_settings (disabled_ml_settings )
335
358
@reset_core_stats_engine ()
336
- @override_application_settings ({"machine_learning.enabled" : False })
337
359
def test_does_not_include_events_when_machine_learning_enabled_is_false ():
338
360
@validate_ml_event_count (count = 0 )
339
361
@background_task ()
0 commit comments