Added tests of place detector. Fixed #5.

WebarchivCZ · Jul 19, 2015 · 75fc617 · 75fc617
1 parent 92c6735
commit 75fc617
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 2 deletions.
diff --git a/tests/analyzers/test_language_detector.py b/tests/analyzers/test_language_detector.py
@@ -4,8 +4,6 @@
 # Interpreter version: python 2.7
 #
 # Imports =====================================================================
-import pytest
-
 from wa_kat.analyzers.language_detector import get_lang_tags
 
 

diff --git a/tests/analyzers/test_place_detector.py b/tests/analyzers/test_place_detector.py
@@ -0,0 +1,34 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Interpreter version: python 2.7
+#
+# Imports =====================================================================
+from wa_kat.analyzers.place_detector import get_place_tags
+
+
+# Variables ===================================================================
+TEST_TEMPLATE = """
+<HTML>
+<head>
+    <title>HTML title</title>
+
+    <meta name    = "geo.placename"
+          content = "Praha">
+</head>
+<body>
+Somecontent.
+</body>
+</HTML>
+"""
+
+
+# Tests =======================================================================
+def test_get_place_tags():
+    place_tags = get_place_tags(TEST_TEMPLATE, "nkp.cz")
+
+    assert place_tags[0] == "Praha"
+    assert place_tags[0].source == "HTML"
+
+    assert place_tags[1] == "Narodni knihovna Ceske republiky, Klementinum 190, Praha 1, 110 00, The Czech Republic"
+    assert place_tags[1].source == "Whois"