Linting

Author: NoahStapp

tests/performance/manage.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 """Django's command-line utility for administrative tasks."""
+
 import os
 import sys
 

tests/performance/perftest/models.py

@@ -1,8 +1,9 @@
 from django.db import models
 
-from django_mongodb_backend.fields import EmbeddedModelField, EmbeddedModelArrayField, ObjectIdField
+from django_mongodb_backend.fields import EmbeddedModelArrayField, EmbeddedModelField, ObjectIdField
 from django_mongodb_backend.models import EmbeddedModel
 
+
 class SmallFlatModel(models.Model):
     field1 = models.CharField(max_length=100)
     field2 = models.CharField(max_length=100)
@@ -18,9 +19,11 @@ class SmallFlatModel(models.Model):
     field12 = models.IntegerField()
     field13 = models.IntegerField()
 
+
 class ForeignKeyModel(models.Model):
     name = models.CharField(max_length=100)
 
+
 class SmallFlatModelFk(models.Model):
     field1 = models.CharField(max_length=100)
     field2 = models.CharField(max_length=100)
@@ -37,6 +40,7 @@ class SmallFlatModelFk(models.Model):
     field13 = models.IntegerField()
     field_fk = models.ForeignKey(ForeignKeyModel, on_delete=models.DO_NOTHING)
 
+
 class LargeFlatModel(models.Model):
     field1 = models.CharField(max_length=100)
     field2 = models.CharField(max_length=100)
@@ -308,6 +312,7 @@ class StringEmbeddedModel(EmbeddedModel):
     field14 = models.CharField(max_length=100)
     field15 = models.CharField(max_length=100)
 
+
 class IntegerEmbeddedModel(EmbeddedModel):
     unique_id = ObjectIdField()
     field1 = models.IntegerField()
@@ -326,6 +331,7 @@ class IntegerEmbeddedModel(EmbeddedModel):
     field14 = models.IntegerField()
     field15 = models.IntegerField()
 
+
 class LargeNestedModel(models.Model):
     embedded_str_doc_1 = EmbeddedModelField(StringEmbeddedModel)
     embedded_str_doc_2 = EmbeddedModelField(StringEmbeddedModel)

tests/performance/perftest/settings.py

@@ -30,12 +30,9 @@
 
 # Application definition
 
-INSTALLED_APPS = [
-    "perftest"
-]
+INSTALLED_APPS = ["perftest"]
 
-MIDDLEWARE = [
-]
+MIDDLEWARE = []
 
 TEMPLATES = [
     {
@@ -107,4 +104,4 @@
 # Default primary key field type
 # https://docs.djangoproject.com/en/5.2/ref/settings/#default-auto-field
 
-DEFAULT_AUTO_FIELD = "django_mongodb_backend.fields.ObjectIdAutoField"
+DEFAULT_AUTO_FIELD = "django_mongodb_backend.fields.ObjectIdAutoField"

tests/performance/perftest/tests.py

@@ -2,14 +2,21 @@
 import os
 import time
 import warnings
-from typing import List
 
-from bson import encode, ObjectId
+from bson import ObjectId, encode
 from django.test import (
     TestCase,
 )
 
-from .models import SmallFlatModel, ForeignKeyModel, SmallFlatModelFk, LargeFlatModel, LargeNestedModel, StringEmbeddedModel, IntegerEmbeddedModel
+from .models import (
+    ForeignKeyModel,
+    IntegerEmbeddedModel,
+    LargeFlatModel,
+    LargeNestedModel,
+    SmallFlatModel,
+    SmallFlatModelFk,
+    StringEmbeddedModel,
+)
 
 OUTPUT_FILE = os.environ.get("OUTPUT_FILE")
 
@@ -18,7 +25,8 @@
 MAX_ITERATION_TIME = 60
 NUM_DOCS = 10000
 
-result_data: List = []
+result_data: list = []
+
 
 def tearDownModule():
     output = json.dumps(result_data, indent=4)
@@ -28,6 +36,7 @@ def tearDownModule():
     else:
         print(output)
 
+
 class Timer:
     def __enter__(self):
         self.start = time.monotonic()
@@ -37,6 +46,7 @@ def __exit__(self, *args):
         self.end = time.monotonic()
         self.interval = self.end - self.start
 
+
 # Copied from the driver benchmarking suite.
 class PerformanceTest:
     dataset: str
@@ -88,9 +98,8 @@ def percentile(self, percentile):
             sorted_results = sorted(self.results)
             percentile_index = int(len(sorted_results) * percentile / 100) - 1
             return sorted_results[percentile_index]
-        else:
-            self.fail("Test execution failed")
-            return None
+        self.fail("Test execution failed")
+        return None
 
     def runTest(self):
         results = []
@@ -123,12 +132,13 @@ class SmallFlatDocTest(PerformanceTest):
 
     def setUp(self):
         super().setUp()
-        with open(self.dataset, "r") as data:
+        with open(self.dataset) as data:
             self.document = json.load(data)
 
         self.data_size = len(encode(self.document)) * NUM_DOCS
         self.documents = [self.document.copy() for _ in range(NUM_DOCS)]
 
+
 class TestSmallFlatDocCreation(SmallFlatDocTest, TestCase):
     def do_task(self):
         for doc in self.documents:
@@ -138,6 +148,7 @@ def do_task(self):
     def after(self):
         SmallFlatModel.objects.all().delete()
 
+
 class TestSmallFlatDocUpdate(SmallFlatDocTest, TestCase):
     def setUp(self):
         super().setUp()
@@ -155,6 +166,7 @@ def do_task(self):
     def after(self):
         SmallFlatModel.objects.all().delete()
 
+
 class TestSmallFlatDocFilterById(SmallFlatDocTest, TestCase):
     def setUp(self):
         super().setUp()
@@ -172,6 +184,7 @@ def tearDown(self):
         super().tearDown()
         SmallFlatModel.objects.all().delete()
 
+
 class TestSmallFlatDocFilterByForeignKey(SmallFlatDocTest, TestCase):
     def setUp(self):
         super().setUp()
@@ -192,17 +205,19 @@ def tearDown(self):
         super().tearDown()
         SmallFlatModelFk.objects.all().delete()
 
+
 class LargeFlatDocTest(PerformanceTest):
     dataset = "large_doc.json"
 
     def setUp(self):
         super().setUp()
-        with open(self.dataset, "r") as data:
+        with open(self.dataset) as data:
             self.document = json.load(data)
 
         self.data_size = len(encode(self.document)) * NUM_DOCS
         self.documents = [self.document.copy() for _ in range(NUM_DOCS)]
 
+
 class TestLargeFlatDocCreation(LargeFlatDocTest, TestCase):
     def do_task(self):
         for doc in self.documents:
@@ -212,6 +227,7 @@ def do_task(self):
     def after(self):
         LargeFlatModel.objects.all().delete()
 
+
 class TestLargeFlatDocUpdate(LargeFlatDocTest, TestCase):
     def setUp(self):
         super().setUp()
@@ -229,12 +245,13 @@ def do_task(self):
     def after(self):
         LargeFlatModel.objects.all().delete()
 
+
 class LargeNestedDocTest(PerformanceTest):
     dataset = "large_doc_nested.json"
 
     def setUp(self):
         super().setUp()
-        with open(self.dataset, "r") as data:
+        with open(self.dataset) as data:
             self.document = json.load(data)
 
         self.data_size = len(encode(self.document)) * NUM_DOCS
@@ -261,13 +278,15 @@ def create_model(self):
                     setattr(model, k, embedded_int_model)
             model.save()
 
+
 class TestLargeNestedDocCreation(LargeNestedDocTest, TestCase):
     def do_task(self):
         self.create_model()
 
     def after(self):
         LargeNestedModel.objects.all().delete()
 
+
 class TestLargeNestedDocUpdate(LargeNestedDocTest, TestCase):
     def setUp(self):
         super().setUp()
@@ -283,11 +302,14 @@ def do_task(self):
             model.embedded_str_doc_1.field1 = "updated_value"
             model.save()
 
+
 class TestLargeNestedDocFilterById(LargeNestedDocTest, TestCase):
     def setUp(self):
         super().setUp()
         self.create_model()
-        self.ids = [model.embedded_str_doc_1.unique_id for model in list(LargeNestedModel.objects.all())]
+        self.ids = [
+            model.embedded_str_doc_1.unique_id for model in list(LargeNestedModel.objects.all())
+        ]
 
     def do_task(self):
         for _id in self.ids:
@@ -297,11 +319,15 @@ def tearDown(self):
         super().tearDown()
         LargeNestedModel.objects.all().delete()
 
+
 class TestLargeNestedDocFilterArray(LargeNestedDocTest, TestCase):
     def setUp(self):
         super().setUp()
         self.create_model()
-        self.ids = [model.embedded_str_doc_array[0].unique_id for model in list(LargeNestedModel.objects.all())]
+        self.ids = [
+            model.embedded_str_doc_array[0].unique_id
+            for model in list(LargeNestedModel.objects.all())
+        ]
 
     def do_task(self):
         for _id in self.ids:
@@ -310,5 +336,3 @@ def do_task(self):
     def tearDown(self):
         super().tearDown()
         LargeNestedModel.objects.all().delete()
-
-