Updated tests

ankane · ankane · commit a715c6834661 · 2023-07-31T21:09:42.000-07:00
diff --git a/tests/test_django.py b/tests/test_django.py
@@ -136,27 +136,21 @@ def test_serialization(self):
                     obj.save()
 
     def test_form(self):
-        form = ItemForm(data={'embedding': "[1, 2, 3]"})
+        form = ItemForm(data={'embedding': '[1, 2, 3]'})
         assert form.is_valid()
         assert 'value="[1, 2, 3]"' in form.as_div()
 
-    def test_form_has_changed(self):
-        Item(id=1, embedding=[1, 2, 3]).save()
-        item = Item.objects.get(pk=1)
-        form = ItemForm(instance=item, data={'embedding': "[1, 2, 4]"})
-        assert form.has_changed()
-
     def test_form_instance(self):
         Item(id=1, embedding=[1, 2, 3]).save()
         item = Item.objects.get(pk=1)
         form = ItemForm(instance=item)
-        # TODO fix
-        # assert 'value="[1.0, 2.0, 3.0]"' in form.as_div()
+        assert 'value="[1.0, 2.0, 3.0]"' in form.as_div()
 
     def test_form_save(self):
         Item(id=1, embedding=[1, 2, 3]).save()
         item = Item.objects.get(pk=1)
-        form = ItemForm(instance=item, data={'embedding': "[4, 5, 6]"})
+        form = ItemForm(instance=item, data={'embedding': '[4, 5, 6]'})
         assert form.is_valid()
         assert form.save()
+        assert form.has_changed()
         assert [4, 5, 6] == Item.objects.get(pk=1).embedding.tolist()
