python-docs-samples/language/classify_text/classify_text_tutorial_test.py at master · ackul/python-docs-samples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Copyright 2016, Google, Inc.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os

import pytest

import classify_text_tutorial


OUTPUT = 'index.json'
RESOURCES = os.path.join(os.path.dirname(__file__), 'resources')
QUERY_TEXT = """Google Home enables users to speak voice commands to interact
with services through the Home\'s intelligent personal assistant called
Google Assistant. A large number of services, both in-house and third-party,
are integrated, allowing users to listen to music, look at videos or photos,
or receive news updates entirely by voice."""
QUERY_CATEGORY = '/Computers & Electronics/Software'


@pytest.fixture(scope='session')
def index_file(tmpdir_factory):
    temp_file = tmpdir_factory.mktemp('tmp').join(OUTPUT)
    temp_out = temp_file.strpath
    classify_text_tutorial.index(os.path.join(RESOURCES, 'texts'), temp_out)
    return temp_file


def test_classify(capsys):
    with open(os.path.join(RESOURCES, 'query_text1.txt'), 'r') as f:
        text = f.read()
    classify_text_tutorial.classify(text)
    out, err = capsys.readouterr()
    assert 'category' in out


def test_index(capsys, tmpdir):
    temp_dir = tmpdir.mkdir('tmp')
    temp_out = temp_dir.join(OUTPUT).strpath

    classify_text_tutorial.index(os.path.join(RESOURCES, 'texts'), temp_out)
    out, err = capsys.readouterr()

    assert OUTPUT in out
    assert len(temp_dir.listdir()) == 1


def test_query_text(capsys, index_file):
    temp_out = index_file.strpath

    classify_text_tutorial.query(temp_out, QUERY_TEXT)
    out, err = capsys.readouterr()

    assert 'Filename: cloud_computing.txt' in out


def test_query_category(capsys, index_file):
    temp_out = index_file.strpath

    classify_text_tutorial.query_category(temp_out, QUERY_CATEGORY)
    out, err = capsys.readouterr()

    assert 'Filename: cloud_computing.txt' in out


def test_split_labels():
    categories = {'/a/b/c': 1.0}
    split_categories = {'a': 1.0, 'b': 1.0, 'c': 1.0}
    assert classify_text_tutorial.split_labels(categories) == split_categories


def test_similarity():
    empty_categories = {}
    categories1 = {'/a/b/c': 1.0, '/d/e': 1.0}
    categories2 = {'/a/b': 1.0}

    assert classify_text_tutorial.similarity(
        empty_categories, categories1) == 0.0
    assert classify_text_tutorial.similarity(categories1, categories1) > 0.99
    assert classify_text_tutorial.similarity(categories1, categories2) > 0
    assert classify_text_tutorial.similarity(categories1, categories2) < 1