Skip to content

Instantly share code, notes, and snippets.

@T-Dimov

T-Dimov/sample_test.py

Last active Apr 9, 2016
Embed
What would you like to do?
import datetime
import unittest
from time import sleep
import solution
class TestSocialGraph(unittest.TestCase):
def setUp(self):
self.terry = solution.User("Terry Gilliam")
self.eric = solution.User("Eric Idle")
self.graham = solution.User("Graham Chapman")
self.john = solution.User("John Cleese")
self.michael = solution.User("Michael Palin")
self.graph = solution.SocialGraph()
self.graph.add_user(self.terry)
self.graph.add_user(self.eric)
self.graph.add_user(self.graham)
self.graph.add_user(self.john)
self.graph.add_user(self.michael)
def test_add_get_and_delete_user(self):
with self.assertRaises(solution.UserAlreadyExistsError):
self.graph.add_user(self.terry)
self.graph.delete_user(self.terry.uuid)
with self.assertRaises(solution.UserDoesNotExistError):
self.graph.get_user(self.terry.uuid)
with self.assertRaises(solution.UserDoesNotExistError):
self.graph.delete_user(self.terry.uuid)
self.graph.add_user(self.terry)
self.assertEqual(self.graph.get_user(self.terry.uuid), self.terry)
def test_following(self):
self.graph.follow(self.terry.uuid, self.eric.uuid)
self.assertTrue(
self.graph.is_following(self.terry.uuid, self.eric.uuid))
self.assertFalse(
self.graph.is_following(self.eric.uuid, self.terry.uuid))
self.assertTrue(
self.graph.following(self.terry.uuid) == {self.eric.uuid})
self.assertTrue(
self.graph.followers(self.eric.uuid) == {self.terry.uuid})
def test_friends(self):
self.graph.follow(self.terry.uuid, self.eric.uuid)
self.assertNotIn(self.eric.uuid, self.graph.friends(self.terry.uuid))
self.assertNotIn(self.terry.uuid, self.graph.friends(self.eric.uuid))
self.graph.follow(self.eric.uuid, self.terry.uuid)
self.assertIn(self.eric.uuid, self.graph.friends(self.terry.uuid))
self.assertIn(self.terry.uuid, self.graph.friends(self.eric.uuid))
def test_distnaces(self):
self.graph.follow(self.terry.uuid, self.eric.uuid)
self.graph.follow(self.terry.uuid, self.graham.uuid)
self.graph.follow(self.eric.uuid, self.michael.uuid)
self.graph.follow(self.eric.uuid, self.john.uuid)
self.graph.follow(self.john.uuid, self.graham.uuid)
self.assertEqual(self.graph.max_distance(self.terry.uuid), 2)
self.assertEqual(self.graph.max_distance(self.eric.uuid), 2)
self.assertEqual(
self.graph.min_distance(self.terry.uuid, self.graham.uuid), 1)
self.assertEqual(
self.graph.min_distance(self.eric.uuid, self.graham.uuid), 2)
def test_layer_followings(self):
self.graph.follow(self.eric.uuid, self.terry.uuid)
self.graph.follow(self.terry.uuid, self.john.uuid)
self.assertEqual(
list(self.graph.nth_layer_followings(self.eric.uuid, 2)),
[self.john.uuid])
def test_feed(self):
self.graph.follow(self.terry.uuid, self.eric.uuid)
self.graph.follow(self.terry.uuid, self.graham.uuid)
self.graph.follow(self.terry.uuid, self.john.uuid)
self.graph.follow(self.terry.uuid, self.michael.uuid)
for i in range(10):
self.eric.add_post(str(i))
sleep(0.000001)
self.graham.add_post(str(10 + i))
sleep(0.000001)
self.john.add_post(str(20 + i))
sleep(0.000001)
self.michael.add_post(str(30 + i))
sleep(0.000001)
self.assertEqual(
[post.content
for post in self.graph.generate_feed(self.terry.uuid, 0, 10)],
["39", "29", "19", "9", "38", "28", "18", "8", "37", "27"])
self.assertEqual(
[post.content
for post in self.graph.generate_feed(self.terry.uuid, 10, 10)],
["17", "7", "36", "26", "16", "6", "35", "25", "15", "5"])
self.assertEqual(
[post.content
for post in self.graph.generate_feed(self.terry.uuid, 20, 10)],
["34", "24", "14", "4", "33", "23", "13", "3", "32", "22"])
self.assertEqual(
[post.content
for post in self.graph.generate_feed(self.terry.uuid, 30, 10)],
["12", "2", "31", "21", "11", "1", "30", "20", "10", "0"])
self.assertEqual([post.content for post in self.eric.get_post()],
[str(i) for i in range(10)])
self.assertEqual([post.content for post in self.graham.get_post()],
[str(i) for i in range(10, 20)])
self.assertEqual([post.content for post in self.john.get_post()],
[str(i) for i in range(20, 30)])
self.assertEqual([post.content for post in self.michael.get_post()],
[str(i) for i in range(30, 40)])
class TestUser(unittest.TestCase):
def setUp(self):
self.michael = solution.User("Michael Palin")
self.terry = solution.User("Terry Gilliam")
def test_has_uuid(self):
self.assertIsNotNone(getattr(self.michael, 'uuid'))
def test_add_post(self):
self.michael.add_post("larodi")
post = next(self.michael.get_post())
self.assertEqual(post.author, self.michael.uuid)
self.assertEqual(post.content, "larodi")
self.assertTrue(isinstance(post.published_at, datetime.datetime))
for i in range(102):
self.terry.add_post("spam")
count = 0
for post in self.terry.get_post():
self.assertEqual(post.content, "spam")
count += 1
self.assertEqual(count, 50)
if __name__ == '__main__':
unittest.main()
@T-Dimov

This comment has been minimized.

Copy link
Owner Author

@T-Dimov T-Dimov commented Apr 9, 2016

Modification of this.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.