From 9aeba6bdc4eb0c4fda0a6ac399fc4f92de2e05c1 Mon Sep 17 00:00:00 2001
From: mlbonhomme <bonhomme@teklia.com>
Date: Thu, 21 Mar 2024 15:59:10 +0100
Subject: [PATCH] Add dataset sets to processes, no datasets
---
arkindex/training/tests/test_datasets_api.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/arkindex/training/tests/test_datasets_api.py b/arkindex/training/tests/test_datasets_api.py
index 9d6da08146..75f027c259 100644
--- a/arkindex/training/tests/test_datasets_api.py
+++ b/arkindex/training/tests/test_datasets_api.py
@@ -684,7 +684,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
def test_update_ponos_task_state_requires_dataset_in_process(self):
self.process.process_sets.all().delete()
- with self.assertNumQueries(5):
+ with self.assertNumQueries(4):
response = self.client.put(
reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),
HTTP_AUTHORIZATION=f"Ponos {self.task.token}",
@@ -845,7 +845,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
def test_partial_update_ponos_task_state_requires_dataset_in_process(self):
self.process.process_sets.all().delete()
- with self.assertNumQueries(5):
+ with self.assertNumQueries(4):
response = self.client.patch(
reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),
HTTP_AUTHORIZATION=f"Ponos {self.task.token}",
@@ -1076,7 +1076,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
self.client.force_login(self.user)
# Remove dataset sets from process
ProcessDatasetSet.objects.filter(process_id=self.process.id, set__dataset_id=self.dataset.id).delete()
- with self.assertNumQueries(7):
+ with self.assertNumQueries(8):
response = self.client.delete(
reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),
)
@@ -1116,7 +1116,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
ProcessDatasetSet.objects.filter(process_id=self.process.id, set__dataset_id=self.dataset.id).delete()
self.client.force_login(self.user)
- with self.assertNumQueries(7):
+ with self.assertNumQueries(8):
response = self.client.delete(
reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),
)
--
GitLab