test.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-u
from pylinkedin.scraper import *
from pylinkedin.utils import *
import unittest
import pytest
import time
import codecs

def read_linkedin_template(file_name):
    with codecs.open('data/' + file_name,'r',encoding = 'utf-8') as f:
        data = f.read()
    return data

def clock(func):
    """ decorator to measure the duration of each test of the unittest suite,
    this is extensible for any kind of functions it will just add a print  """
    def clocked(*args, **kwargs):
        t0 = time.time()
        result = func(*args)
        elapsed = (time.time() - t0) * 1000  # in ms
        print ('elapsed : [{0:0.3f}ms]'.format(elapsed))
        return result


class TestUtils(unittest.TestCase):

    def test_extract_one(self):
        assert extract_one([u'engineer'])  == u'engineer'
        assert extract_one([]) is None

    def test_clean(self):
        assert clean(u'  engineer ') == u'engineer'
        assert clean([' R',' Python',' Matlab']) == [u'R', u'Python', u'Matlab']
        assert clean(3) == 3
        assert clean(3.0) == 3.0



class TestLinkedinItem(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.l = LinkedinItem(html_string = read_linkedin_template('jeff_weiner.txt'))

    def test_url_detected(self):
        assert self.l.url_detected == 'https://www.linkedin.com/in/jeffweiner08'

    # def test_number_connections(self):
    #     assert self.l.number_connections == u'500+'

    def test_name(self):
        assert self.l.name == u'Jeff Weiner'

    def test_first_name(self):
        assert self.l.first_name == u'Jeff'

    def test_last_name(self):
        assert self.l.last_name == u'Weiner'

    def test_to_dict(self):
        assert isinstance(self.l.to_dict(),dict)

    def test_current_title(self):
        assert self.l.current_title == u'CEO at LinkedIn'

    # def test_current_location(self):
    #     assert self.l.current_location == u'Région de la baie de San Francisco, États-Unis'

    def test_current_industry(self):
        assert self.l.current_industry == u'Internet'


# class TestLinkedinOffline(unittest.TestCase):

#     def setUp(self):
#         """ creating test data set for the test module """
#         self.linkedin = LinkedinItem(html_string = read_linkedin_template('test_adarshmani.html'))

#     def test_get_name(self):
#         assert self.linkedin.name == u'Adarsh Kumar Mani'

#     def test_get_current_title(self):
#         assert self.linkedin.current_title == u'Software Engineer at Apple'

#     def test_get_current_education(self):
#         assert isinstance(self.linkedin.current_education,dict) == True
#         assert self.linkedin.current_education['name'] == u'University of California, Berkeley'
#         assert self.linkedin.current_education['url'] == u'http://www.linkedin.com/edu/school?id=17939&trk=ppro_sprof'