Add tests for NLP example

[Anmol-Srivastava]

Task

• unit tests for added components
• verify output and results in example notebook, adding plots where relevant