spacy-js/tests/span.test.js

JavaScript API for spaCy with Python REST API

import spacy from '../src'
import { Doc, Token, Span } from '../src/tokens'
import { text, words, spaces, attrs } from './util'

jest.mock('../src/language')

const nlp = spacy.load('en_core_web_sm')

test('allows manual construction', async () => {
    const doc = await nlp(text)
    const span = new Span(doc, 6, 9)
    expect(span).toBeInstanceOf(Span)
    expect(span.text).toBe('sentence about Facebook')
})

test('allows being sliced off Doc', async () => {
    const doc = await nlp(text)
    const span = doc.slice(6, 9)
    expect(span).toBeInstanceOf(Span)
    expect(span.text).toBe('sentence about Facebook')
})

test('has Span attributes', async () => {
    const doc = await nlp(text)
    const span = doc.slice(6, 9)
    expect(span.toString()).toBe('sentence about Facebook')
    expect(span.length).toBe(3)
    expect(span.start).toBe(6)
    expect(span.end).toBe(9)
    expect(span.label).toBeUndefined()
})

test('has parent Doc', async () => {
    const doc = await nlp(text)
    const span = doc.slice(6, 9)
    expect(span.doc).toBeInstanceOf(Doc)
    expect(span.doc).toBe(doc)
})

test('has entity label', async () => {
    const doc = await nlp(text)
    const span = doc.slice(8, 9)
    expect(span.toString()).toBe('Facebook')
    expect(span.label).toBe('ORG')
})

test('allows token indexing', async () => {
    const doc = await nlp(text)
    const span = doc.slice(6, 9)
    for (let i = 0; i < span.length; i++) {
        expect(span[i]).toBeInstanceOf(Token)
    }
    expect(span[span.length + 1]).toBeUndefined()
})

test('allows token iteration', async () => {
    const doc = await nlp(text)
    const span = doc.slice(6, 9)
    for (let token of span) {
        expect(token).toBeInstanceOf(Token)
    }
})