|
|
|
|
|
from __future__ import unicode_literals |
|
from nose.tools import eq_ |
|
from pytube import utils |
|
|
|
|
|
class TestUtils(object): |
|
blob = "Lorem ipsum dolor sit amet, consectetur adipiscing elit" |
|
|
|
def test_truncate(self): |
|
"""Truncate string works as expected""" |
|
truncated = utils.truncate(self.blob, 11) |
|
eq_(truncated, 'Lorem ipsum') |
|
|
|
def test_safe_filename(self): |
|
"""Unsafe characters get stripped from generated filename""" |
|
eq_(utils.safe_filename("abc1245$$"), "abc1245") |
|
eq_(utils.safe_filename("abc##"), "abc") |
|
eq_(utils.safe_filename("abc:foo"), "abc -foo") |
|
eq_(utils.safe_filename("abc_foo"), "abc foo") |
|
|
|
def test_sizeof(self): |
|
"""Accurately converts the bytes to its humanized equivalent""" |
|
eq_(utils.sizeof(1), '1 byte') |
|
eq_(utils.sizeof(2), '2 bytes') |
|
eq_(utils.sizeof(2400), '2 KB') |
|
eq_(utils.sizeof(2400000), '2 MB') |
|
eq_(utils.sizeof(2400000000), '2 GB') |
|
eq_(utils.sizeof(2400000000000), '2 TB') |
|
eq_(utils.sizeof(2400000000000000), '2 PB') |
|
|
