22from datetime import datetime , timedelta
33from unittest import mock
44
5+ from hamcrest import assert_that , is_ , equal_to
6+
57from flagsmith .analytics import ANALYTICS_TIMER , AnalyticsProcessor
68
79
@@ -10,18 +12,18 @@ def test_analytics_processor_track_feature_updates_analytics_data(
1012) -> None :
1113 # When
1214 analytics_processor .track_feature ("my_feature" )
13- assert analytics_processor .analytics_data ["my_feature" ] == 1
15+ assert_that ( analytics_processor .analytics_data ["my_feature" ], is_ ( 1 ))
1416
1517 analytics_processor .track_feature ("my_feature" )
16- assert analytics_processor .analytics_data ["my_feature" ] == 2
18+ assert_that ( analytics_processor .analytics_data ["my_feature" ], is_ ( 2 ))
1719
1820
1921def test_analytics_processor_flush_clears_analytics_data (
2022 analytics_processor : AnalyticsProcessor ,
2123) -> None :
2224 analytics_processor .track_feature ("my_feature" )
2325 analytics_processor .flush ()
24- assert analytics_processor .analytics_data == {}
26+ assert_that ( analytics_processor .analytics_data , equal_to ({}))
2527
2628
2729def test_analytics_processor_flush_post_request_data_match_ananlytics_data (
@@ -36,7 +38,10 @@ def test_analytics_processor_flush_post_request_data_match_ananlytics_data(
3638 # Then
3739 session .post .assert_called ()
3840 post_call = session .mock_calls [0 ]
39- assert {"my_feature_1" : 1 , "my_feature_2" : 1 } == json .loads (post_call [2 ]["data" ])
41+ assert_that (
42+ json .loads (post_call [2 ]["data" ]),
43+ equal_to ({"my_feature_1" : 1 , "my_feature_2" : 1 }),
44+ )
4045
4146
4247def test_analytics_processor_flush_early_exit_if_analytics_data_is_empty (
0 commit comments