Advertisement
Guest User

Untitled

a guest
Aug 24th, 2017
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.89 KB | None | 0 0
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3.  
  4.  
  5. def mask_digits(x, y):
  6. """mask_digits
  7.  
  8. Mask the higher digits of a number with those of a larger number
  9.  
  10. :param x: number as str or int
  11. :param y: number as str or int
  12. :return x, y: tuple of the original numbers expanded
  13. """
  14. if y is not None:
  15. n_x = str(x)
  16. n_y = str(y)
  17. if len(n_x) > len(n_y):
  18. diff = len(n_x) - len(n_y)
  19. n_y = n_x[0:diff] + n_y
  20.  
  21. x = int(n_x)
  22. y = int(n_y)
  23.  
  24. return (x, y)
  25.  
  26.  
  27. def test_none():
  28. assert mask_digits(66234, None) == (66234, None)
  29.  
  30.  
  31. def test_mask_no_digits():
  32. assert mask_digits(11, 15) == (11, 15)
  33.  
  34.  
  35. def test_mask_4_digits():
  36. assert mask_digits(12345, 6) == (12345, 12346)
  37.  
  38.  
  39. def test_mask_3_digits():
  40. assert mask_digits(12345, 68) == (12345, 12368)
  41.  
  42.  
  43. def test_mask_1_digits():
  44. assert mask_digits(8492, 512) == (8492, 8512)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement