From fe844148f6490cd281ff5feb530fdd8941fa1091 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Tue, 17 Oct 2017 19:43:52 +0200
Subject: [PATCH] Test pickling hooks

---
 spacy/tests/doc/test_pickle_doc.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/spacy/tests/doc/test_pickle_doc.py b/spacy/tests/doc/test_pickle_doc.py
index 2571f30ac..93f06f2c3 100644
--- a/spacy/tests/doc/test_pickle_doc.py
+++ b/spacy/tests/doc/test_pickle_doc.py
@@ -41,3 +41,14 @@ def test_user_data_unpickles():
     b = pickle.dumps(doc)
     doc2 = pickle.loads(b)
     assert doc2.user_data[(0, 1)] == False
+
+
+def test_hooks_unpickle():
+    def inner_func(d1, d2):
+        return 'hello!'
+    nlp = Language()
+    doc = nlp(u'Hello')
+    doc.user_hooks['similarity'] = inner_func
+    b = pickle.dumps(doc)
+    doc2 = pickle.loads(b)
+    assert doc2.similarity(None) == 'hello!'