Skip to content

Commit b7bc8ca

Browse files
committed
Improved typing for Django tests [skip ci]
1 parent 95cd718 commit b7bc8ca

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

tests/test_django.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -345,13 +345,13 @@ def test_serialization(self) -> None:
345345
def test_vector_form(self) -> None:
346346
form = VectorForm(data={'embedding': '[1, 2, 3]'})
347347
assert form.is_valid()
348-
assert 'value="[1, 2, 3]"' in form.as_div() # type: ignore
348+
assert 'value="[1, 2, 3]"' in str(form.as_div())
349349

350350
def test_vector_form_instance(self) -> None:
351351
Item(id=1, embedding=[1, 2, 3]).save()
352352
item = Item.objects.get(pk=1)
353353
form = VectorForm(instance=item)
354-
assert 'value="[1.0, 2.0, 3.0]"' in form.as_div() # type: ignore
354+
assert 'value="[1.0, 2.0, 3.0]"' in str(form.as_div())
355355

356356
def test_vector_form_save(self) -> None:
357357
Item(id=1, embedding=[1, 2, 3]).save()
@@ -373,13 +373,13 @@ def test_vector_form_save_missing(self) -> None:
373373
def test_halfvec_form(self) -> None:
374374
form = HalfVectorForm(data={'half_embedding': '[1, 2, 3]'})
375375
assert form.is_valid()
376-
assert 'value="[1, 2, 3]"' in form.as_div() # type: ignore
376+
assert 'value="[1, 2, 3]"' in str(form.as_div())
377377

378378
def test_halfvec_form_instance(self) -> None:
379379
Item(id=1, half_embedding=[1, 2, 3]).save()
380380
item = Item.objects.get(pk=1)
381381
form = HalfVectorForm(instance=item)
382-
assert 'value="[1.0, 2.0, 3.0]"' in form.as_div() # type: ignore
382+
assert 'value="[1.0, 2.0, 3.0]"' in str(form.as_div())
383383

384384
def test_halfvec_form_save(self) -> None:
385385
Item(id=1, half_embedding=[1, 2, 3]).save()
@@ -401,13 +401,13 @@ def test_halfvec_form_save_missing(self) -> None:
401401
def test_bit_form(self) -> None:
402402
form = BitForm(data={'binary_embedding': '101'})
403403
assert form.is_valid()
404-
assert 'value="101"' in form.as_div() # type: ignore
404+
assert 'value="101"' in str(form.as_div())
405405

406406
def test_bit_form_instance(self) -> None:
407407
Item(id=1, binary_embedding='101').save()
408408
item = Item.objects.get(pk=1)
409409
form = BitForm(instance=item)
410-
assert 'value="101"' in form.as_div() # type: ignore
410+
assert 'value="101"' in str(form.as_div())
411411

412412
def test_bit_form_save(self) -> None:
413413
Item(id=1, binary_embedding='101').save()
@@ -429,14 +429,14 @@ def test_bit_form_save_missing(self) -> None:
429429
def test_sparsevec_form(self) -> None:
430430
form = SparseVectorForm(data={'sparse_embedding': '{1:1,2:2,3:3}/3'})
431431
assert form.is_valid()
432-
assert 'value="{1:1,2:2,3:3}/3"' in form.as_div() # type: ignore
432+
assert 'value="{1:1,2:2,3:3}/3"' in str(form.as_div())
433433

434434
def test_sparsevec_form_instance(self) -> None:
435435
Item(id=1, sparse_embedding=[1, 2, 3]).save()
436436
item = Item.objects.get(pk=1)
437437
form = SparseVectorForm(instance=item)
438438
# TODO improve
439-
assert 'value="{1:1.0,2:2.0,3:3.0}/3"' in form.as_div() # type: ignore
439+
assert 'value="{1:1.0,2:2.0,3:3.0}/3"' in str(form.as_div())
440440

441441
def test_sparsevec_form_save(self) -> None:
442442
Item(id=1, sparse_embedding=[1, 2, 3]).save()

0 commit comments

Comments
 (0)