diff options
Diffstat (limited to '')
| -rw-r--r-- | ctrack/organisations/tests/test_models.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/ctrack/organisations/tests/test_models.py b/ctrack/organisations/tests/test_models.py index cb3b9af..740b782 100644 --- a/ctrack/organisations/tests/test_models.py +++ b/ctrack/organisations/tests/test_models.py @@ -1,12 +1,12 @@ import pytest from slugify import slugify -from ..models import Organisation, Address +from ctrack.organisations.models import Organisation pytestmark = pytest.mark.django_db -def test_organisation_get_absolute_url(org: Organisation): +def test_organisation_get_absolute_url(org): slug = slugify(org.name) assert org.get_absolute_url() == f"/organisations/{slug}/" @@ -17,7 +17,7 @@ def test_create_organisation(): assert Organisation.objects.get(name="Big Bad OES Corporation") -def test_delete_organisation(org: Organisation): +def test_delete_organisation(org): orgs = Organisation.objects.all() assert org in orgs Organisation.delete(org) @@ -25,13 +25,13 @@ def test_delete_organisation(org: Organisation): assert Organisation.objects.count() == 0 -def test_update_organisation(org: Organisation): +def test_update_organisation(org): # Change the name of the organisation org.name = "Tonkers Ltd" org.save() assert org.name == "Tonkers Ltd" -def test_new_address(addr: Address): +def test_new_address(addr): # The address "has" an organisation assert addr.organisation.name |