add no coerce perf test

Author: bckohan

django_enum/tests/enum_prop/migrations/0001_initial.py

@@ -1,7 +1,7 @@
-# Generated by Django 3.2.14 on 2022-07-26 03:42
+# Generated by Django 3.2.14 on 2022-08-09 03:11
 
-import django_enum.fields
 from django.db import migrations, models
+import django_enum.fields
 
 
 class Migration(migrations.Migration):
@@ -85,4 +85,11 @@ class Migration(migrations.Migration):
                 ('small_pos_int', models.PositiveSmallIntegerField(blank=True, choices=[(0, 'Value 1'), (2, 'Value 2'), (32767, 'Value 32767')], db_index=True, default=None, null=True)),
             ],
         ),
+        migrations.CreateModel(
+            name='SingleNoCoercePerf',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('small_pos_int', django_enum.fields.EnumPositiveSmallIntegerField(blank=True, choices=[(0, 'Value 1'), (2, 'Value 2'), (32767, 'Value 32767')], db_index=True, default=None, null=True)),
+            ],
+        ),
     ]

django_enum/tests/enum_prop/models.py

@@ -182,5 +182,10 @@ class SingleFieldPerf(models.Model):
 
         small_pos_int = models.PositiveSmallIntegerField(choices=SmallPosIntEnum.choices, null=True, default=None, db_index=True, blank=True)
 
+
+    class SingleNoCoercePerf(models.Model):
+
+        small_pos_int = EnumField(enum=SmallPosIntEnum, coerce=False, null=True, default=None, db_index=True, blank=True)
+
 except (ImportError, ModuleNotFoundError):  # pragma: no cover
     pass

django_enum/tests/tests.py

@@ -1154,6 +1154,7 @@ def test_bulk_update(self):
         PerfCompare,
         SingleEnumPerf,
         SingleFieldPerf,
+        SingleNoCoercePerf
     )
     from enum_properties import EnumProperties, s
 
@@ -2674,12 +2675,19 @@ def test_single_field_perf_diff(self):
                 SingleFieldPerf.objects.create(small_pos_int=0)
             choice_stop = perf_counter()
 
+            no_coerce_start = perf_counter()
+            for idx in range(0, self.COUNT):
+                SingleNoCoercePerf.objects.create(small_pos_int=0)
+            no_coerce_end = perf_counter()
+
             enum_time = enum_stop - enum_start
             choice_time = choice_stop - choice_start
+            no_coerce_time = no_coerce_end - no_coerce_start
 
+            print(f'{enum_time} {choice_time} {no_coerce_time}')
             # Enum tends to be about ~12% slower
             self.assertTrue((enum_time / choice_time) < 1.5)
-            # print(f'{enum_time} {choice_time}')
+            self.assertTrue((no_coerce_time / choice_time) < 1.1)
 
             enum_start = perf_counter()
             for _ in SingleEnumPerf.objects.all():
@@ -2691,13 +2699,22 @@ def test_single_field_perf_diff(self):
                 continue
             choice_stop = perf_counter()
 
+            no_coerce_start = perf_counter()
+            for _ in SingleNoCoercePerf.objects.all():
+                continue
+            no_coerce_end = perf_counter()
+
             enum_time = enum_stop - enum_start
             choice_time = choice_stop - choice_start
+            no_coerce_time = no_coerce_end - no_coerce_start
 
-            # print(f'{enum_time} {choice_time}')
+            print(f'{enum_time} {choice_time} {no_coerce_time}')
             # tends to be about 1.8x slower
             self.assertTrue((enum_time / choice_time) < 2.5)
 
+            # todo how to tighten this? why so much slower still?
+            self.assertTrue((no_coerce_time / choice_time) < 1.7)
+
 
     class ExampleTests(TestCase):  # pragma: no cover  - why is this necessary?