import test from 'tape'
import nlp from '../../lib/two.js'
const here = '[two/output/out] '

test('out-tags', function (t) {
  const out = nlp(`he's cool.`).out('tags')
  t.equal(out.length, 1, 'one sentence')
  t.equal(out[0].he.indexOf('Pronoun') !== -1, true, here + 'has he:Pronoun')
  t.equal(out[0].is.indexOf('Copula') !== -1, true, here + 'has is:Copula')
  t.equal(out[0].cool.indexOf('Adjective') !== -1, true, here + 'has cool:Adjective')
  t.end()
})

test('out-topk', function (t) {
  const doc = nlp(`What'd I say? Monorail. and it called? Monorail.`)
  const out = doc.out('freq')
  // t.equal(out.length, 3, here + 'three results')
  t.equal(out[0].normal, 'monorail.', here + 'top-result is most-frequent')
  t.equal(out[0].count, 2, here + 'got count')
  t.end()
})

test('out-array', function (t) {
  const arr = nlp('1-2').terms().out('array')
  t.equal(arr.length, 1, here + 'one result')
  t.equal(arr[0], '1-2', here + 'got contraction text')
  t.end()
})

test('out-wrap', function (t) {
  let doc = nlp("soft and yielding like a nerf ball")
  let out = doc.out({
    '#Adjective': (m) => `[${m.text()}]`
  })
  t.equal(out, `[soft] and [yielding] like a nerf ball`, here + 'two matches')

  // pre-post
  doc = nlp("before (match) after")
  out = doc.wrap({
    'match': () => `few more words`,
  })
  t.equal(out, `before (few more words) after`, here + 'pre+post')
  t.end()
})

test('out-wrap-punctuation', function (t) {
  const arr = [
    [`"hello world" he said`, `"[hello world]" he said`],
    [`he said "hello world" today`, `he said "[hello world]" today`],
    [`he said "hello world"`, `he said "[hello world]"`],
    [`(hello world) he said`, `([hello world]) he said`],
    [`he said (hello world) today`, `he said ([hello world]) today`],
    [`he said (hello world).`, `he said ([hello world]).`],
    [`hello world, he said`, `[hello world], he said`],
    [`he left. "hello world" she said`, `he left. "[hello world]" she said`],
    [`"hello world."`, `"[hello world]."`],
  ]
  arr.forEach(function (a) {
    const doc = nlp(a[0])
    t.equal(doc.out({ 'hello world': m => `[${m.text()}]` }), a[1], here + 'text: ' + a[0])
    t.equal(doc.wrap({ 'hello world': m => `[${m.out('normal')}]` }), a[1], here + 'normal: ' + a[0])
  })
  const out = nlp(`"Hello World" he said`).wrap({ 'hello world': m => `[${m.text('normal')}]` })
  t.equal(out, `"[hello world]" he said`, here + 'keeps text format')
  t.end()
})