Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python
- # -*- coding: utf-8 -*-
- def mask_digits(x, y):
- """mask_digits
- Mask the higher digits of a number with those of a larger number
- :param x: number as str or int
- :param y: number as str or int
- :return x, y: tuple of the original numbers expanded
- """
- if y is not None:
- n_x = str(x)
- n_y = str(y)
- if len(n_x) > len(n_y):
- diff = len(n_x) - len(n_y)
- n_y = n_x[0:diff] + n_y
- x = int(n_x)
- y = int(n_y)
- return (x, y)
- def test_none():
- assert mask_digits(66234, None) == (66234, None)
- def test_mask_no_digits():
- assert mask_digits(11, 15) == (11, 15)
- def test_mask_4_digits():
- assert mask_digits(12345, 6) == (12345, 12346)
- def test_mask_3_digits():
- assert mask_digits(12345, 68) == (12345, 12368)
- def test_mask_1_digits():
- assert mask_digits(8492, 512) == (8492, 8512)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement