test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Detective Yellowcopyrightedrat - A Telegram bot to organize Pokémon GO raids
# Copyright (C) 2017 Jorge Suárez de Lis <hey@gentakojima.me>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

from supportmethods import parse_profile_image
import re
import os
import argparse
import random
import time
import sys
import cv2
import logging


logging.basicConfig(filename='/dev/stdout', filemode='w', format='%(asctime)s %(message)s', level=logging.DEBUG)

parser = argparse.ArgumentParser()
parser.add_argument('--inspect', help='Examine failed images interactively', action='store_true')
args = parser.parse_args()

passed_tests = failed_tests = 0

# Load test files
for root, dirs, filenames in os.walk("testingimgs"):
    if root != "testingimgs":
        continue
    for f in filenames:
        m = re.match(r'(Rojo|Amarillo|Azul)_([0-9]{2})_([A-Za-z0-9]{3,15})_([A-Za-z]{3,12})_([A-Za-z0-9]{3,12})\.jpg', f)
        if m is None:
            print("[!] Ignoring file %s..." % f)
        else:
            print("[*] Testing file %s..." % f)

            expected_color = m.group(1)
            expected_level = m.group(2)
            expected_trainer_name = m.group(3)
            expected_pokemon = m.group(4)
            expected_pokemon_name = m.group(5)

            inspectFilename = str(time.time()) + "_" + str(random.randrange(100, 999))
            (trainer_name, level, chosen_color, chosen_pokemon, pokemon_name, chosen_profile) = parse_profile_image(os.path.join(root, f), expected_pokemon, inspect=True, inspectFilename=inspectFilename)

            failed_test = False

            try:
                assert chosen_color == expected_color
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect team! '%s' vs. expected '%s'" % (chosen_color, expected_color))
                failed_tests = failed_tests + 1
                failed_test = True
            try:
                assert level == expected_level
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect level! '%s' vs. expected '%s'" % (level, expected_level))
                failed_tests = failed_tests + 1
                failed_test = True
            try:
                assert trainer_name.replace("I","l").replace("1","l").lower() == expected_trainer_name.replace("1","l").lower()
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect trainer name! '%s' vs. expected '%s'" % (trainer_name, expected_trainer_name))
                failed_tests = failed_tests + 1
                failed_test = True
            try:
                assert chosen_pokemon == expected_pokemon
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect Pokémon! '%s' vs. expected '%s'" % (chosen_pokemon, expected_pokemon))
                failed_tests = failed_tests + 1
                failed_test = True
            try:
                assert pokemon_name.lower() == expected_pokemon_name.lower()
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect Pokémon name! '%s' vs. expected '%s'" % (pokemon_name, expected_pokemon_name))
                failed_tests = failed_tests + 1
                failed_test = True
            try:
                assert chosen_profile is not None
                passed_tests = passed_tests + 1
            except AssertionError as e:
                print(" [!] Incorrect Profile model! No match at all!")
                failed_tests = failed_tests + 1
                failed_test = True

            if failed_test and args.inspect:
                inspectdir = sys.path[0] + "/inspectimages"
                for dirname, dirnames, filenames in os.walk(inspectdir):
                    for f in filenames:
                        if re.match(r'%s' % inspectFilename, f) is not None:
                            i = cv2.imread(inspectdir + "/" + f)
                            cv2.imshow("Inspecting %s" % f, i)
                cv2.waitKey(0)

print("Passed tests: %s" % passed_tests)
print("Failed tests: %s" % failed_tests)