From 687da825e8c60516e4948742238274f385875acc Mon Sep 17 00:00:00 2001 From: "M1ha.Shvn" Date: Wed, 19 Oct 2022 12:51:57 +0500 Subject: [PATCH] Fix in tests on issue-42 --- tests/test_models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index c8f3a2c..1fa2011 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -107,7 +107,7 @@ class TestOperations(TransactionTestCase): ], returning='*') self.assertEqual(5, len(res)) - for i, instance in enumerate(items): + for i, instance in enumerate(res): self.assertEqual(instance.created, now_dt) self.assertEqual(instance.created_date, now_dt.date()) self.assertEqual(i, instance.value) @@ -140,7 +140,7 @@ class TestOperations(TransactionTestCase): ], returning='*') self.assertEqual(2, len(res)) - for instance in items: + for instance in res: self.assertEqual(instance.value, instance.pk * 10) self.assertSetEqual({('update', "%s.%d" % (self.db_alias, instance.pk)) for instance in items}, @@ -182,7 +182,7 @@ class TestOperations(TransactionTestCase): for instance in res: self.assertEqual(instance.value, instance.pk * 10) - self.assertSetEqual({('update', "%s.%d" % (self.db_alias, instance.pk)) for instance in items}, + self.assertSetEqual({('update', "%s.%d" % (self.db_alias, instance.pk)) for instance in res}, set(self.storage.get_operations(self.clickhouse_model.get_import_key(), 10))) def test_get_or_create(self):