1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# -*- coding: utf-8 -*-
#
# Copyright (C) 2019 Radim Rehurek <me@radimrehurek.com>
#
# This code is distributed under the terms and conditions
# from the MIT License (MIT).
#
import urllib.parse
import pytest
import smart_open.utils
@pytest.mark.parametrize(
'value,minval,maxval,expected',
[
(5, 0, 10, 5),
(11, 0, 10, 10),
(-1, 0, 10, 0),
(10, 0, None, 10),
(-10, 0, None, 0),
]
)
def test_clamp(value, minval, maxval, expected):
assert smart_open.utils.clamp(value, minval=minval, maxval=maxval) == expected
@pytest.mark.parametrize(
'value,params,expected',
[
(10, {}, 10),
(-10, {}, 0),
(-10, {'minval': -5}, -5),
(10, {'maxval': 5}, 5),
]
)
def test_clamp_defaults(value, params, expected):
assert smart_open.utils.clamp(value, **params) == expected
def test_check_kwargs():
import smart_open.s3
kallable = smart_open.s3.open
kwargs = {'client': 'foo', 'unsupported': 'bar', 'client_kwargs': 'boaz'}
supported = smart_open.utils.check_kwargs(kallable, kwargs)
assert supported == {'client': 'foo', 'client_kwargs': 'boaz'}
@pytest.mark.parametrize(
'url,expected',
[
('s3://bucket/key', ('s3', 'bucket', '/key', '', '')),
('s3://bucket/key?', ('s3', 'bucket', '/key?', '', '')),
('s3://bucket/???', ('s3', 'bucket', '/???', '', '')),
('https://host/path?foo=bar', ('https', 'host', '/path', 'foo=bar', '')),
]
)
def test_safe_urlsplit(url, expected):
actual = smart_open.utils.safe_urlsplit(url)
assert actual == urllib.parse.SplitResult(*expected)
|
