# coding: utf-8
from __future__ import unicode_literals

from spacy.tokens import Doc
from spacy.attrs import ORTH, SHAPE, POS, DEP

from ..util import get_doc


def test_doc_array_attr_of_token(en_vocab):
    doc = Doc(en_vocab, words=["An", "example", "sentence"])
    example = doc.vocab["example"]
    assert example.orth != example.shape
    feats_array = doc.to_array((ORTH, SHAPE))
    assert feats_array[0][0] != feats_array[0][1]
    assert feats_array[0][0] != feats_array[0][1]


def test_doc_stringy_array_attr_of_token(en_vocab):
    doc = Doc(en_vocab, words=["An", "example", "sentence"])
    example = doc.vocab["example"]
    assert example.orth != example.shape
    feats_array = doc.to_array((ORTH, SHAPE))
    feats_array_stringy = doc.to_array(("ORTH", "SHAPE"))
    assert feats_array_stringy[0][0] == feats_array[0][0]
    assert feats_array_stringy[0][1] == feats_array[0][1]


def test_doc_scalar_attr_of_token(en_vocab):
    doc = Doc(en_vocab, words=["An", "example", "sentence"])
    example = doc.vocab["example"]
    assert example.orth != example.shape
    feats_array = doc.to_array(ORTH)
    assert feats_array.shape == (3,)


def test_doc_array_tag(en_vocab):
    words = ["A", "nice", "sentence", "."]
    pos = ["DET", "ADJ", "NOUN", "PUNCT"]
    doc = get_doc(en_vocab, words=words, pos=pos)
    assert doc[0].pos != doc[1].pos != doc[2].pos != doc[3].pos
    feats_array = doc.to_array((ORTH, POS))
    assert feats_array[0][1] == doc[0].pos
    assert feats_array[1][1] == doc[1].pos
    assert feats_array[2][1] == doc[2].pos
    assert feats_array[3][1] == doc[3].pos


def test_doc_array_dep(en_vocab):
    words = ["A", "nice", "sentence", "."]
    deps = ["det", "amod", "ROOT", "punct"]
    doc = get_doc(en_vocab, words=words, deps=deps)
    feats_array = doc.to_array((ORTH, DEP))
    assert feats_array[0][1] == doc[0].dep
    assert feats_array[1][1] == doc[1].dep
    assert feats_array[2][1] == doc[2].dep
    assert feats_array[3][1] == doc[3].dep

@@ -50,16 +50,30 @@
 terals%0A%0A
+import pytest%0A
 from spa
@@ -2003,12 +2003,442 @@
  doc%5B3%5D.dep%0A
+%0A%0A@pytest.mark.xfail%0A@pytest.mark.parametrize(%22attrs%22, %5B%5B%22ORTH%22, %22SHAPE%22%5D, %22IS_ALPHA%22%5D)%0Adef test_doc_array_to_from_string_attrs(en_vocab, attrs):%0A    %22%22%22Test that both Doc.to_array and Doc.from_array accept string attrs,%0A    as well as single attrs and sequences of attrs.%0A    %22%22%22%0A    words = %5B%22An%22, %22example%22, %22sentence%22%5D%0A    doc = Doc(en_vocab, words=words)%0A    Doc(en_vocab, words=words).from_array(attrs, doc.to_array(attrs))%0A

Add xfailing test for to_array/from_array string attrs