forked from brunobord/neigefr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtests.py
82 lines (76 loc) · 3.22 KB
/
tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
from django.test import TestCase
from neigefr.flake import parse_body, process
from neigefr.models import Snowflake
class FlakeTest(TestCase):
def test_parse_body(self):
content = "#neigefr 64100 2/10"
flake = parse_body(content)
self.assertEquals(flake.zipcode, '64100')
self.assertEquals(flake.ranking, 2)
content = "#neigefr 64100 10/10"
flake = parse_body(content)
self.assertEquals(flake.ranking, 10)
def test_process(self):
data = {
"iso_language_code": "fr",
"to_user_name": "Seb Solere",
"to_user_id_str": "137994749",
"profile_image_url_https": "https://si0.twimg.com/profile_images/329883768/3697028883_0ae17360f9_b_normal.jpg",
"from_user_id_str": "14979950",
"text": "#neigefr 64100 2/10",
"from_user_name": "Bruno Bord",
"in_reply_to_status_id_str": "165073316988723201",
"profile_image_url": "http://a3.twimg.com/profile_images/329883768/3697028883_0ae17360f9_b_normal.jpg",
"id": 165074248593969152,
"to_user": "sebsolere",
"source": "<a href="http://twirssi.com" rel="nofollow">Twirssi</a>",
"in_reply_to_status_id": 165073316988723201,
"id_str": "165074248593969152",
"from_user": "brunobord",
"from_user_id": 14979950,
"to_user_id": 137994749,
"geo": None,
"created_at": "Thu, 02 Feb 2012 14:09:02 +0000",
"metadata": {
"result_type": "recent"
}
}
snowflake = process(data)
zipcode = snowflake.zipcode
self.assertEquals(zipcode.zipcode, '64100')
self.assertEquals(zipcode.city, u'Bayonne, Aquitaine')
self.assertEquals(zipcode.latitude, '43.4833333')
self.assertEquals(zipcode.longitude, '-1.4833333')
self.assertEquals(snowflake.tweet_id, 165074248593969152L)
self.assertEquals(snowflake.latitude, zipcode.latitude)
self.assertEquals(snowflake.longitude, zipcode.longitude)
self.assertEquals(snowflake.rank, 2)
def test_flakesize(self):
snowflake = Snowflake()
snowflake.rank = 0
self.assertEquals(snowflake.flakesize, 16)
snowflake.rank = 1
self.assertEquals(snowflake.flakesize, 16)
snowflake.rank = 2
self.assertEquals(snowflake.flakesize, 20)
snowflake.rank = 3
self.assertEquals(snowflake.flakesize, 20)
snowflake.rank = 4
self.assertEquals(snowflake.flakesize, 24)
snowflake.rank = 5
self.assertEquals(snowflake.flakesize, 24)
snowflake.rank = 6
self.assertEquals(snowflake.flakesize, 28)
snowflake.rank = 7
self.assertEquals(snowflake.flakesize, 28)
snowflake.rank = 8
self.assertEquals(snowflake.flakesize, 32)
snowflake.rank = 9
self.assertEquals(snowflake.flakesize, 32)
snowflake.rank = 10
self.assertEquals(snowflake.flakesize, 32)
# special cases
snowflake.rank = None
self.assertEquals(snowflake.flakesize, 16)
snowflake.rank = 11
self.assertEquals(snowflake.flakesize, 32)