Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: tests/test_rpy.py

Issue 29988581: Issue 7230 - python-abp line2dict vectorization (Closed) Base URL: https://hg.adblockplus.org/python-abp/
Patch Set: Created Jan. 24, 2019, 3:37 a.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« abp/filters/rpy.py ('K') | « abp/filters/rpy.py ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # This file is part of Adblock Plus <https://adblockplus.org/>, 1 # This file is part of Adblock Plus <https://adblockplus.org/>,
2 # Copyright (C) 2006-present eyeo GmbH 2 # Copyright (C) 2006-present eyeo GmbH
3 # 3 #
4 # Adblock Plus is free software: you can redistribute it and/or modify 4 # Adblock Plus is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License version 3 as 5 # it under the terms of the GNU General Public License version 3 as
6 # published by the Free Software Foundation. 6 # published by the Free Software Foundation.
7 # 7 #
8 # Adblock Plus is distributed in the hope that it will be useful, 8 # Adblock Plus is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # GNU General Public License for more details. 11 # GNU General Public License for more details.
12 # 12 #
13 # You should have received a copy of the GNU General Public License 13 # You should have received a copy of the GNU General Public License
14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
15 15
16 """Functional tests for testing rPython integration.""" 16 """Functional tests for testing rPython integration."""
17 from __future__ import unicode_literals 17 from __future__ import unicode_literals
18 18
19 from collections import namedtuple 19 from collections import namedtuple
20 import pytest 20 import pytest
21 import sys 21 import sys
22 22
23 from abp.filters.rpy import tuple2dict, line2dict 23 from abp.filters.rpy import tuple2dict, line2dict, lines2dict
24 24
25 25
26 _SAMPLE_TUPLE = namedtuple('tuple', 'foo,bar') 26 _SAMPLE_TUPLE = namedtuple('tuple', 'foo,bar')
27 27
28 _TEST_EXAMPLES = { 28 _TEST_EXAMPLES = {
29 'header': { 29 'header': {
30 'in': b'[Adblock Plus 2.0]', 30 'in': b'[Adblock Plus 2.0]',
31 'out': { 31 'out': {
32 b'type': b'Header', 32 b'type': b'Header',
33 b'version': b'Adblock Plus 2.0', 33 b'version': b'Adblock Plus 2.0',
(...skipping 106 matching lines...) Expand 10 before | Expand all | Expand 10 after
140 @pytest.mark.parametrize('line_type', list(_TEST_EXAMPLES.keys())) 140 @pytest.mark.parametrize('line_type', list(_TEST_EXAMPLES.keys()))
141 def test_line2dict_format(line_type): 141 def test_line2dict_format(line_type):
142 """Test that the API result has the appropriate format. 142 """Test that the API result has the appropriate format.
143 143
144 Checks for both keys and datatypes. 144 Checks for both keys and datatypes.
145 """ 145 """
146 position = 'start' if line_type in {'header', 'metadata'} else 'body' 146 position = 'start' if line_type in {'header', 'metadata'} else 'body'
147 data = line2dict(_TEST_EXAMPLES[line_type]['in'], position) 147 data = line2dict(_TEST_EXAMPLES[line_type]['in'], position)
148 148
149 assert data == _TEST_EXAMPLES[line_type]['out'] 149 assert data == _TEST_EXAMPLES[line_type]['out']
150
151
152 def test_lines2dict():
153 """Test that the API result has the appropriate format.
154
155 Checks for both keys and datatypes.
156 """
157 string_list = []
158 line_types = {b'EmptyLine': 'empty', b'Filter': 'filter_single',
159 b'Comment': 'comment', b'Include': 'include',
160 b'Header': 'header', b'Metadata': 'metadata'}
161
162 for key in _TEST_EXAMPLES.keys():
163 string_list.append(_TEST_EXAMPLES[key]['in'])
164
165 result_dict = lines2dict(string_list)
166
167 for line in result_dict.keys():
168 line_type = line_types[result_dict[line][b'type']]
169 data = line2dict(_TEST_EXAMPLES[line_type]['in'], 'start')
170
171 assert data == _TEST_EXAMPLES[line_type]['out']
OLDNEW
« abp/filters/rpy.py ('K') | « abp/filters/rpy.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld