# coding: utf8 from __future__ import unicode_literals import unittest import dotdot from pubs.query import (AuthorFilter, CitekeyFilter, FieldFilter, YearFilter, _query_block_to_filter, get_paper_filter, InvalidQuery) from pubs.paper import Paper import fixtures doe_paper = Paper.from_bibentry(fixtures.doe_bibentry) page_paper = Paper.from_bibentry(fixtures.page_bibentry) turing_paper = Paper.from_bibentry(fixtures.turing_bibentry, metadata=fixtures.turing_metadata) class TestAuthorFilter(unittest.TestCase): def test_fails_if_no_author(self): no_doe = doe_paper.deepcopy() no_doe.bibentry['Doe2013']['author'] = [] self.assertFalse(AuthorFilter('whatever')(no_doe)) def test_match_case(self): self.assertTrue(AuthorFilter('doe')(doe_paper)) self.assertTrue(AuthorFilter('doe', case_sensitive=False)(doe_paper)) self.assertTrue(AuthorFilter('Doe')(doe_paper)) def test_do_not_match_case(self): self.assertFalse(AuthorFilter('dOe')(doe_paper)) self.assertFalse(AuthorFilter('dOe', case_sensitive=True)(doe_paper)) self.assertFalse(AuthorFilter('doe', case_sensitive=True)(doe_paper)) self.assertTrue(AuthorFilter('dOe', case_sensitive=False)(doe_paper)) def test_match_not_first_author(self): self.assertTrue(AuthorFilter('motwani')(page_paper)) def test_do_not_match_first_name(self): self.assertFalse(AuthorFilter('lawrence')(page_paper)) class TestCitekeyFilter(unittest.TestCase): def test_fails_if_no_citekey(self): no_citekey = doe_paper.deepcopy() no_citekey.citekey = '' self.assertFalse(CitekeyFilter('whatever')(no_citekey)) def test_match_case(self): self.assertTrue(CitekeyFilter('doe201')(doe_paper)) self.assertTrue(CitekeyFilter('doe201', case_sensitive=False)(doe_paper)) self.assertTrue(CitekeyFilter('Doe201')(doe_paper)) def test_do_not_match_case(self): self.assertFalse(CitekeyFilter('dOe201')(doe_paper)) self.assertFalse(CitekeyFilter('dOe201', case_sensitive=True)(doe_paper)) self.assertFalse(CitekeyFilter('doe201', case_sensitive=True)(doe_paper)) self.assertTrue(CitekeyFilter('dOe201', case_sensitive=False)(doe_paper)) def test_latex_enc(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.citekey = "{G}r{\\\"u}n2013" self.assertTrue(CitekeyFilter('Grün')(latexenc_paper)) self.assertTrue(CitekeyFilter('Gr{\\\"u}n')(latexenc_paper)) def test_normalize_unicode(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.citekey = "Jalape\u00f1o2013" self.assertTrue(CitekeyFilter("Jalapen\u0303o")(latexenc_paper)) def test_strict(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.citekey = "Jalape\u00f1o2013" self.assertFalse(CitekeyFilter("Jalapen\u0303o", strict=True)(latexenc_paper)) latexenc_paper.citekey = "{G}ros2013" self.assertFalse(CitekeyFilter("Gros", strict=True)(latexenc_paper)) def test_strict_implies_case(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.citekey = "Gros2013" self.assertFalse( CitekeyFilter("gros", case_sensitive=False, strict=True)(latexenc_paper)) class TestCheckTag(unittest.TestCase): pass class TestCheckYear(unittest.TestCase): def test_single_year(self): self.assertTrue(YearFilter('2013')(doe_paper)) self.assertFalse(YearFilter('2014')(doe_paper)) def test_before_year(self): self.assertTrue(YearFilter('-2013')(doe_paper)) self.assertTrue(YearFilter('-2014')(doe_paper)) self.assertFalse(YearFilter('-2012')(doe_paper)) def test_after_year(self): self.assertTrue(YearFilter('2013-')(doe_paper)) self.assertTrue(YearFilter('2012-')(doe_paper)) self.assertFalse(YearFilter('2014-')(doe_paper)) def test_year_range(self): self.assertTrue(YearFilter('')(doe_paper)) self.assertTrue(YearFilter('-')(doe_paper)) self.assertTrue(YearFilter('2013-2013')(doe_paper)) self.assertTrue(YearFilter('2012-2014')(doe_paper)) self.assertFalse(YearFilter('2014-2015')(doe_paper)) with self.assertRaises(ValueError): YearFilter('2015-2014')(doe_paper) class TestCheckField(unittest.TestCase): def test_match_case(self): self.assertTrue(FieldFilter('title', 'nice')(doe_paper)) self.assertTrue( FieldFilter('title', 'nice', case_sensitive=False)(doe_paper)) self.assertTrue(FieldFilter('year', '2013')(doe_paper)) def test_do_not_match_case(self): self.assertTrue( FieldFilter('title', 'Title', case_sensitive=True)(doe_paper)) self.assertFalse( FieldFilter('title', 'nice', case_sensitive=True)(doe_paper)) def test_latex_enc(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = "{G}r{\\\"u}n" self.assertTrue( FieldFilter('title', 'Grün')(latexenc_paper)) self.assertTrue( FieldFilter('title', 'Gr{\\\"u}n')(latexenc_paper)) def test_normalize_unicode(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = "Jalape\u00f1o" self.assertTrue( FieldFilter('title', "Jalapen\u0303o")(latexenc_paper)) def test_strict(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = "Jalape\u00f1o" self.assertFalse(FieldFilter('title', "Jalapen\u0303o", strict=True)(latexenc_paper)) latexenc_paper.bibentry['Doe2013']['title'] = "{G}ros" self.assertFalse( FieldFilter('title', "Gros", strict=True)(latexenc_paper)) def test_strict_implies_case(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = "Gros" self.assertFalse( FieldFilter('title', "gros", case_sensitive=False, strict=True)(latexenc_paper)) class TestCheckQueryBlock(unittest.TestCase): def test_raise_invalid_if_no_value(self): with self.assertRaises(InvalidQuery): _query_block_to_filter('title') def test_raise_invalid_if_too_much(self): with self.assertRaises(InvalidQuery): _query_block_to_filter('whatever:value:too_much') class TestFilterPaper(unittest.TestCase): def test_case(self): self.assertTrue(get_paper_filter(['title:nice'])(doe_paper)) self.assertTrue(get_paper_filter(['title:Nice'])(doe_paper)) self.assertFalse(get_paper_filter(['title:nIce'])(doe_paper)) def test_fields(self): self.assertTrue(get_paper_filter(['year:2013'])(doe_paper)) self.assertTrue(get_paper_filter(['year:2010-'])(doe_paper)) self.assertFalse(get_paper_filter(['year:2014'])(doe_paper)) self.assertTrue(get_paper_filter(['author:doe'])(doe_paper)) self.assertTrue(get_paper_filter(['author:Doe'])(doe_paper)) def test_tags(self): self.assertTrue(get_paper_filter(['tag:computer'])(turing_paper)) self.assertFalse(get_paper_filter(['tag:Ai'])(turing_paper)) self.assertTrue(get_paper_filter(['tag:AI'])(turing_paper)) self.assertTrue(get_paper_filter(['tag:ai'])(turing_paper)) def test_multiple(self): self.assertTrue(get_paper_filter(['author:doe', 'year:2013'])(doe_paper)) self.assertTrue(get_paper_filter(['author:doe', 'year:2010-2014'])(doe_paper)) self.assertFalse(get_paper_filter(['author:doe', 'year:2014-'])(doe_paper)) self.assertFalse(get_paper_filter(['author:doee', 'year:2014'])(doe_paper)) def test_latex_enc(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = r"{E}l Ni{\~n}o" latexenc_paper.bibentry['Doe2013']['author'][0] = r"Erd\H{o}s, Paul" self.assertTrue(get_paper_filter(['title:El'])(latexenc_paper)) self.assertTrue(get_paper_filter(['title:Niño'])(latexenc_paper)) self.assertTrue(get_paper_filter(['author:erdős'])(latexenc_paper)) self.assertTrue(get_paper_filter(['title:{E}l'])(latexenc_paper)) def test_normalize_unicode(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = r"{E}l Ni{\~n}o" self.assertTrue(get_paper_filter(['title:Nin\u0303o'])(latexenc_paper)) def test_strict(self): latexenc_paper = doe_paper.deepcopy() latexenc_paper.bibentry['Doe2013']['title'] = r"El Ni{\~n}o" self.assertFalse(get_paper_filter( ['title:Nin\u0303o'], strict=True)(latexenc_paper)) if __name__ == '__main__': unittest.main()