ó
ßüÚ\c           @   sl   d  d l  Z d  d l m Z d  d l Z e j d ƒ Z d  d l m Z m	 Z	 m
 Z
 m Z m Z d „  Z d S(   iÿÿÿÿN(   t   assert_allcloset   keras(   t	   serializet   deserializet   dumpst   loadst   to_serializec          C   s  t  j j ƒ  }  |  j t  j j d d d ƒƒ |  j t  j j d ƒ ƒ |  j d d d d ƒ t j j d ƒ } t j j d ƒ } |  j	 | | ƒ t
 t |  ƒ Œ  } t | j | ƒ |  j | ƒ ƒ i t |  ƒ d
 6} t | ƒ } t | ƒ } t | d
 j | ƒ |  j | ƒ ƒ d  S(   Ni   t	   input_dimi   i   t	   optimizert   sgdt   losst   msei   t   model(   i   i   (   i   i   (   R   t   modelst
   Sequentialt   addt   layerst   Denset   compilet   npt   randomt   train_on_batchR   R   R    t   predictR   R   R   (   R   t   xt   yt   loadedt   datat   framest   result(    (    sD   lib/python2.7/site-packages/distributed/protocol/tests/test_keras.pyt    test_serialize_deserialize_model
   s    (   t   numpyR   t   numpy.testingR    t   pytestt   importorskipR   t   distributed.protocolR   R   R   R   R   R   (    (    (    sD   lib/python2.7/site-packages/distributed/protocol/tests/test_keras.pyt   <module>   s
   (