Skip to content

Commit 3df693f

Browse files
committed
add nullable float test for coverage
1 parent 9f213b3 commit 3df693f

File tree

3 files changed

+26
-0
lines changed

3 files changed

+26
-0
lines changed

tests/djenum/enums.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,13 @@ def __hash__(self):
223223
return super().__hash__()
224224

225225

226+
class NullableConstants(Enum):
227+
NONE = None
228+
PI = 3.14159265358979323846264338327950288
229+
e = 2.71828
230+
GOLDEN_RATIO = 1.61803398874989484820458683436563811
231+
232+
226233
class SmallPositiveFlagEnum(IntFlag):
227234
ONE = 2**10
228235
TWO = 2**11

tests/djenum/models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
StrPropsEnum,
3737
TextEnum,
3838
TimeEnum,
39+
NullableConstants,
3940
)
4041

4142

@@ -312,3 +313,7 @@ class CustomPrimitiveTestModel(models.Model):
312313
path = EnumField(PathEnum, primitive=str)
313314

314315
str_props = EnumField(StrPropsEnum, primitive=str)
316+
317+
318+
class TestNullableFloat(models.Model):
319+
nullable_float = EnumField(NullableConstants, default=None, blank=True, null=True)

tests/test_eccentric.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,3 +161,17 @@ def test_custom_primitive(self):
161161
obj3,
162162
CustomPrimitiveTestModel.objects.get(str_props=StrProps("str3")),
163163
)
164+
165+
def test_nullable_float(self):
166+
from tests.djenum.models import TestNullableFloat
167+
from tests.djenum.enums import NullableConstants
168+
169+
obj = TestNullableFloat.objects.create()
170+
self.assertEqual(obj.nullable_float, NullableConstants.NONE)
171+
obj.refresh_from_db()
172+
self.assertEqual(obj.nullable_float, NullableConstants.NONE)
173+
174+
obj.nullable_float = NullableConstants.PI
175+
obj.save()
176+
obj.refresh_from_db()
177+
self.assertEqual(obj.nullable_float, NullableConstants.PI)

0 commit comments

Comments
 (0)