ó
ßüÚ\c           @   sH   d  d l  Z  e  j d ƒ d  d l Z d  d l m Z m Z d „  Z d S(   iÿÿÿÿNt   sklearn(   t	   serializet   deserializec          C   sÀ   t  j j ƒ  }  |  j d d g d d g d d g g d d d g ƒ t |  ƒ \ } } | d d k sk t ‚ t | | ƒ } d d g d d g g } |  j | ƒ | j | ƒ k j ƒ  s¼ t ‚ d  S(   Ni    i   i   t
   serializert   daski   iÿÿÿÿ(	   R    t   linear_modelt   LinearRegressiont   fitR   t   AssertionErrorR   t   predictt   all(   t   estt   headert   framest   est2t   inp(    (    sF   lib/python2.7/site-packages/distributed/protocol/tests/test_sklearn.pyt
   test_basic
   s    4(   t   pytestt   importorskipt   sklearn.linear_modelR    t   distributed.protocolR   R   R   (    (    (    sF   lib/python2.7/site-packages/distributed/protocol/tests/test_sklearn.pyt   <module>   s   