diff --git a/_unittests/ut_helpers/test_cache_helper.py b/_unittests/ut_helpers/test_cache_helper.py index 16b444c5..cf1d5a74 100644 --- a/_unittests/ut_helpers/test_cache_helper.py +++ b/_unittests/ut_helpers/test_cache_helper.py @@ -180,7 +180,7 @@ def test_make_sliding_window_cache(self): ) text = self.string_type(cache, with_shape=True) self.assertEqual( - "SlidingWindowCache(key_cache=#3[T1s4x5x6x7,T1s4x5x6x7,T1s4x5x6x7], " + f"{type(cache).__name__}(key_cache=#3[T1s4x5x6x7,T1s4x5x6x7,T1s4x5x6x7], " "value_cache=#3[T1s4x5x6x7,T1s4x5x6x7,T1s4x5x6x7])", text, )