lambdaofgod
/
github_search
mirror of https://github.com/lambdaofgod/github_search


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
            from github_search import text_preprocessing
import pytest


@pytest.mark.parametrize("url,expected", [
    ('https://github.com/mperham/sidekiq#requirements', 'xxurl #requirements'),
    ('https://github.com/mperham/sidekiq', 'xxurl ')
])
def test_url_replacement(url, expected):
    rule = text_preprocessing.Rules.url_replacement
    cleaned_url = text_preprocessing.replace_with_rules(url, [rule])
    assert cleaned_url == expected


@pytest.mark.parametrize("url,expected", [
    ('#', 'xxhashtag '),
    ('##', 'xxhashtag ')
])
def test_hashtag_replacement(url, expected):
    rule = text_preprocessing.Rules.hashtag_replacement
    cleaned_url = text_preprocessing.replace_with_rules(url, [rule])
    assert cleaned_url == expected