Advertisement
Guest User

Untitled

a guest
Jun 15th, 2019
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.92 KB | None | 0 0
  1. import random
  2. import unittest
  3.  
  4. def get_random_line(filename):
  5. lines = []
  6.  
  7. with open(filename) as f:
  8. for line in f:
  9. lines.append(line)
  10.  
  11. return random.choice(lines)
  12.  
  13.  
  14. class TestRandomLine(unittest.TestCase):
  15. def test_random_line(self):
  16. filename = 'test.txt'
  17. lines = 10
  18.  
  19. with open(filename, 'w') as f:
  20. for i in range(lines):
  21. f.write('line number {0}\n'.format(i))
  22.  
  23. seed = 1
  24. random.seed(seed)
  25. rand_num = random.randrange(lines)
  26. with open(filename) as f:
  27. line1 = ''
  28. for i in range(rand_num + 1):
  29. line1 = f.readline()
  30.  
  31. random.seed(seed)
  32. line2 = get_random_line(filename)
  33. self.assertEqual(line1, line2)
  34.  
  35. random.seed(seed + 1)
  36. line3 = get_random_line(filename)
  37. self.assertNotEqual(line1, line3)
  38.  
  39. if __name__ == '__main__':
  40. unittest.main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement