|
5 | 5 |
|
6 | 6 | class test__regression__cache_on_self(TestCase): |
7 | 7 |
|
| 8 | + def test__regression__cache_on_self__enum_caching(self): |
| 9 | + |
| 10 | + from enum import Enum |
| 11 | + |
| 12 | + @cache_on_self |
| 13 | + def an_method(self, an_str:str): |
| 14 | + return an_str |
| 15 | + |
| 16 | + assert an_method(self, an_str='A') == 'A' |
| 17 | + assert an_method(self, an_str='B') == 'B' |
| 18 | + assert an_method(self, an_str='C') == 'C' |
| 19 | + |
| 20 | + class An_Enum(str, Enum): |
| 21 | + A = 'A' |
| 22 | + B = 'B' |
| 23 | + C = 'C' |
| 24 | + |
| 25 | + @cache_on_self |
| 26 | + def an_method_with_enum(self, an_enum:An_Enum): |
| 27 | + return an_enum |
| 28 | + |
| 29 | + assert an_method_with_enum(self, an_enum='A') == 'A' |
| 30 | + assert an_method_with_enum(self, an_enum='B') == 'B' |
| 31 | + assert an_method_with_enum(self, an_enum='C') == 'C' |
| 32 | + |
| 33 | + assert an_method_with_enum(self, an_enum=An_Enum.A) == 'A' |
| 34 | + assert an_method_with_enum(self, an_enum=An_Enum.A) == An_Enum.A |
| 35 | + #assert an_method_with_enum(self, an_enum=An_Enum.B) != 'B' # BUG |
| 36 | + #assert an_method_with_enum(self, an_enum=An_Enum.B) == An_Enum.A # BUG |
| 37 | + #assert an_method_with_enum(self, an_enum=An_Enum.C) == An_Enum.A # BUG |
| 38 | + assert an_method_with_enum(self, an_enum=An_Enum.B) == An_Enum.B # FIXED |
| 39 | + assert an_method_with_enum(self, an_enum=An_Enum.B) == 'B' # FIXED |
| 40 | + assert an_method_with_enum(self, an_enum=An_Enum.C) == An_Enum.C # FIXED |
8 | 41 | def test__regression__cache_on_self__none_values_ignored(self): |
9 | 42 | """Test that None values are completely ignored in cache key generation""" |
10 | 43 | class None_Values_Class: |
|
0 commit comments