import spacy
import pytest

@pytest.mark.models
def test_beam_parse():
    nlp = spacy.load('en_core_web_sm')
    doc = nlp(u'Australia is a country', disable=['ner'])
    ents = nlp.entity(doc, beam_width=2)
    print(ents)


from __future__ import unicode_literals

import pytest


@pytest.mark.models('en')
def test_beam_parse(EN):
    doc = EN(u'Australia is a country', disable=['ner'])
    ents = EN.entity(doc, beam_width=2)
    print(ents)

Fix beam parse model test
Fix beam parse model test

- import spacy
+ from __future__ import unicode_literals
+ 
  import pytest
  
+ 
- @pytest.mark.models
+ @pytest.mark.models('en')
- def test_beam_parse():
+ def test_beam_parse(EN):
-     nlp = spacy.load('en_core_web_sm')
-     doc = nlp(u'Australia is a country', disable=['ner'])
+     doc = EN(u'Australia is a country', disable=['ner'])
-     ents = nlp.entity(doc, beam_width=2)
+     ents = EN.entity(doc, beam_width=2)
      print(ents)
  
- 
Fix beam parse model test
## Code Before:
import spacy
import pytest

@pytest.mark.models
def test_beam_parse():
    nlp = spacy.load('en_core_web_sm')
    doc = nlp(u'Australia is a country', disable=['ner'])
    ents = nlp.entity(doc, beam_width=2)
    print(ents)


## Instruction:
Fix beam parse model test
## Code After:
from __future__ import unicode_literals

import pytest


@pytest.mark.models('en')
def test_beam_parse(EN):
    doc = EN(u'Australia is a country', disable=['ner'])
    ents = EN.entity(doc, beam_width=2)
    print(ents)
