294 lines
8.3 KiB
Python
294 lines
8.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import datetime
|
|
from django.test import TestCase
|
|
from django.db import models
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.utils import timezone
|
|
|
|
from lostplaces.models import Place, PlaceVoting
|
|
from lostplaces.tests.models import ModelTestCase
|
|
|
|
class PlaceTestCase(ModelTestCase):
|
|
model = Place
|
|
related_name = 'places'
|
|
nullable = True
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
user = User.objects.create_user(
|
|
username='testpeter',
|
|
password='Develop123'
|
|
)
|
|
|
|
place = Place.objects.create(
|
|
name='Im a place',
|
|
submitted_when=timezone.now(),
|
|
submitted_by=user.explorer,
|
|
location='Testtown',
|
|
latitude=50.5,
|
|
longitude=7.0,
|
|
description='This is just a test, do not worry'
|
|
)
|
|
place.tags.add('I a tag', 'testlocation')
|
|
place.save()
|
|
|
|
|
|
def setUp(self):
|
|
self.place = Place.objects.get(id=1)
|
|
|
|
def test_location(self):
|
|
self.assertCharField(
|
|
field_name='location',
|
|
min_length=10,
|
|
max_length=100
|
|
)
|
|
|
|
def test_description(self):
|
|
self.assertField('description', models.TextField)
|
|
|
|
def test_average_latlon(self):
|
|
'''
|
|
Tests the average latitude/longitude calculation of a list
|
|
of 10 places
|
|
'''
|
|
place_list = []
|
|
for i in range(10):
|
|
place = Place.objects.get(id=1)
|
|
place.id = None
|
|
place.latitude = i+1
|
|
place.longitude = i+10
|
|
place.save()
|
|
place_list.append(place)
|
|
|
|
avg_latlon = Place.average_latlon(place_list)
|
|
|
|
self.assertTrue('latitude' in avg_latlon,
|
|
msg='Expecting avg_latlon dict to have an \'latitude\' key'
|
|
)
|
|
self.assertTrue('longitude' in avg_latlon,
|
|
msg='Expecting avg_latlon dict to have an \'longitude\' key'
|
|
)
|
|
|
|
self.assertEqual(avg_latlon['latitude'], 5.5,
|
|
msg='%s: average latitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
self.assertEqual(avg_latlon['longitude'], 14.5,
|
|
msg='%s: average longitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
|
|
def test_average_latlon_one_place(self):
|
|
'''
|
|
Tests the average latitude/longitude calculation of a list
|
|
of one place
|
|
'''
|
|
place = Place.objects.get(id=1)
|
|
avg_latlon = Place.average_latlon([place])
|
|
self.assertEqual(avg_latlon['latitude'], place.latitude,
|
|
msg='%s:(one place) average latitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
self.assertEqual(avg_latlon['longitude'], place.longitude,
|
|
msg='%s: (one place) average longitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
|
|
def test_average_latlon_no_places(self):
|
|
'''
|
|
Tests the average latitude/longitude calculation of
|
|
an empty list
|
|
'''
|
|
avg_latlon = Place.average_latlon([])
|
|
self.assertEqual(avg_latlon['latitude'], 51.961922091398904,
|
|
msg='%s: (no places) average latitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
self.assertEqual(avg_latlon['longitude'], 7.6295628132604385,
|
|
msg='%s: (no places) average longitude missmatch' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
|
|
def test_str(self):
|
|
place = self.place
|
|
self.assertTrue(place.name.lower() in str(place).lower(),
|
|
msg='Expecting %s.__str__ to contain the name' % (
|
|
self.model.__name__
|
|
)
|
|
)
|
|
|
|
def test_level_calculation(self):
|
|
explorer = self.place.submitted_by
|
|
|
|
PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote=5
|
|
)
|
|
PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote=2
|
|
)
|
|
PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote=4
|
|
)
|
|
|
|
self.place.calculate_place_level()
|
|
|
|
self.assertEqual(
|
|
4,
|
|
self.place.level,
|
|
msg='Expecting the place level to be 4'
|
|
)
|
|
|
|
def test_level_calculation_no_votes(self):
|
|
self.place.calculate_place_level()
|
|
self.assertEqual(
|
|
5,
|
|
self.place.level,
|
|
msg='Expecting the default place level to be 5'
|
|
)
|
|
|
|
def test_level_mid_accuracy(self):
|
|
explorer = self.place.submitted_by
|
|
six_month_ago = datetime.timedelta(days=180)
|
|
self.place.submitted_when = timezone.now() - six_month_ago
|
|
|
|
votings = [
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=170),
|
|
'vote': 5
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=23),
|
|
'vote': 2
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=1),
|
|
'vote': 4
|
|
}
|
|
]
|
|
|
|
for vote in votings:
|
|
voting = PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote= vote['vote']
|
|
)
|
|
voting.submitted_when = vote['date']
|
|
voting.save()
|
|
|
|
self.assertEqual(
|
|
65,
|
|
self.place.calculate_voting_accuracy()
|
|
)
|
|
|
|
def test_level_high_accuracy(self):
|
|
explorer = self.place.submitted_by
|
|
six_month_ago = datetime.timedelta(days=180)
|
|
self.place.submitted_when = timezone.now() - six_month_ago
|
|
|
|
votings = [
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=9),
|
|
'vote': 5
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=14),
|
|
'vote': 2
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=0),
|
|
'vote': 4
|
|
}
|
|
]
|
|
|
|
for vote in votings:
|
|
voting = PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote= vote['vote']
|
|
)
|
|
voting.submitted_when = vote['date']
|
|
voting.save()
|
|
|
|
self.assertEqual(
|
|
96,
|
|
self.place.calculate_voting_accuracy()
|
|
)
|
|
|
|
def test_level_low_accuracy(self):
|
|
explorer = self.place.submitted_by
|
|
six_month_ago = datetime.timedelta(days=180)
|
|
self.place.submitted_when = timezone.now() - six_month_ago
|
|
|
|
votings = [
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=177),
|
|
'vote': 5
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=150),
|
|
'vote': 2
|
|
},
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=100),
|
|
'vote': 4
|
|
}
|
|
]
|
|
|
|
for vote in votings:
|
|
voting = PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote= vote['vote']
|
|
)
|
|
voting.submitted_when = vote['date']
|
|
voting.save()
|
|
|
|
self.assertEqual(
|
|
21,
|
|
self.place.calculate_voting_accuracy()
|
|
)
|
|
|
|
def test_level_accuracy_zero_timedelta(self):
|
|
explorer = self.place.submitted_by
|
|
six_month_ago = datetime.timedelta(days=180)
|
|
self.place.submitted_when = timezone.now() - six_month_ago
|
|
|
|
votings = [
|
|
{
|
|
'date': timezone.now() - datetime.timedelta(days=0),
|
|
'vote': 4
|
|
}
|
|
]
|
|
|
|
for vote in votings:
|
|
voting = PlaceVoting.objects.create(
|
|
submitted_by=explorer,
|
|
place=self.place,
|
|
vote= vote['vote']
|
|
)
|
|
voting.submitted_when = vote['date']
|
|
voting.save()
|
|
|
|
self.assertEqual(
|
|
100,
|
|
self.place.calculate_voting_accuracy(),
|
|
msg='Expecting the accurcy to be 100% when the vote is 0 time units old'
|
|
)
|
|
|
|
|