#!/usr/bin/env python
# -*- coding: utf-8 -*-
from wpalign import * 

enlinks = extract_wikitext_links(u"[[Rennes|The city of Rennes]] is in [[France]].")

if  enlinks ==  [u'Rennes', u'France']: 
  print 'extracted english links ok.'


frlinks = extract_wikitext_links(
u"""La 'France', ou la 'République française' pour 
les usages officiels, est un [[pays]] dont la [[France métropolitaine|métropole]] 
se situe en [[Europe de l’Ouest]], et qui possède des 
[[France d'outre-mer|territoires]] à divers endroits du [[Globe terrestre|Globe]]"""
)

if frlinks ==  [u'pays', u'France métropolitaine', u'Europe de l’Ouest', u'France d\'outre-mer', u'Globe terrestre']:
  print 'extracted french links ok.'

english_sentence = Sentence(u"[[Rennes|The city of Rennes]] is in [[France]].")
if  english_sentence.links ==  [u'Rennes', u'France']: 
  print "english sentence right"


french_sentence = Sentence(
u"""La 'France', ou la 'République française' pour 
les usages officiels, est un [[pays]] dont la [[France métropolitaine|métropole]] 
se situe en [[Europe de l’Ouest]], et qui possède des 
[[France d'outre-mer|territoires]] à divers endroits du [[Globe terrestre|Globe]]"""
)

if french_sentence.links ==  [u'pays', u'France métropolitaine', u'Europe de l’Ouest', u'France d\'outre-mer', u'Globe terrestre']:
  print 'french links ok.'