LoginSignup
1
0

More than 1 year has passed since last update.

ソフトウェアテストの小ネタAdvent Calendar 2021Day 8
@pbjpkas

Pythonで始めるテストツール製作

Last updated at Posted at 2021-12-07

1. はじめに

Pythonはライブラリが充実していて、使いやすくてパワフルなテストツールをお手軽に作れます。本稿ではMenu Based CLI(メニュー形式のコマンドラインインターフェース)で操作するテストツールの基本形とそのカスタマイズ例をご紹介します。

例1: 数値計算
例2: 画像の一部を切り出して保存

PythonやOpenCVのインストール方法は付録Aを、動作確認環境は付録Bをご覧ください。

2. 基本形

カスタマイズのベースとなるプログラムを以下に示します。

myTools.py
# myTools.py by ka's(https://qiita.com/pbjpkas) 2021
# python 3.x
import sys


def function_a():
    print("Hello, World.")


def main():
    while True:
        print("== myTools ==")
        print("a: function_a")
        print("x: exit")

        s = input(">")

        if s == "a":
            function_a()

        if s == "x":
            print("Bye.")
            sys.exit()


main()

step.1

Cドライブの直下にworkというフォルダを作成し上記プログラムをmyTools.pyという名前で保存します。
source.png

NOTE:
ソースコードのウィンドウにマウスポインタを乗せると右上にアイコンが出現します。
これをクリックするとソースコードがクリップボードへコピーされます。
qiita-source-copy.png

step.2

次に、コマンドプロンプトを起動しpythonで実行します。

cmd
cd c:\work
python myTools.py

step.3

以下のようにメニューとプロンプト(>)が表示されればOKです。

== myTools ==
a: function_a
x: exit
>
  • aを入力してEnterするとfunction_a()が実行され、メニューに戻ります。
  • xを入力してEnterするとBye.と表示され終了します。
  • それ以外は無視され、メニューが再度表示されます。
実行例
c:\work>python myTools.py
== myTools ==
a: function_a
x: exit
>a
Hello, World.
== myTools ==
a: function_a
x: exit
>b
== myTools ==
a: function_a
x: exit
>x
Bye.

c:\work>

3. 数値計算

一つ目の例として「先月と今月のメトリクス」のような2つの数値の比較を想定します。OSに標準でインストールされている電卓でも計算できますが、「差」と「何倍か」の両方を知りたいといった、計算する項目が複数あるとちょっと手間です。そこでテストツールに組み込みます。

要件

  • 新旧の値を入力すると「差」および「何倍か」を計算すること
    • 旧の値が0の場合は「何倍か」は0とすること
  • 何倍かは小数点二けたで表示すること

実装

  • input()の戻り値は文字列(str型)のためfloat型にキャストして計算します。
myTools.py(Ver.2)
# myTools.py by ka's(https://qiita.com/pbjpkas) 2021
# python 3.x
import sys


def comparison_value():
    try:
        old_value = float(input("old>"))
        new_value = float(input("new>"))
    except:
        print(sys.exc_info())
        return False

    delta = new_value - old_value
    if old_value != 0:
        times = new_value / old_value
    else:
        times = 0

    print("delta:%f, times:%.2f" % (delta, times))
    return True


def main():
    while True:
        print("== myTools ==")
        print("a: comparison of old and new value")
        print("x: exit")

        s = input(">")

        if s == "a":
            comparison_value()

        if s == "x":
            print("Bye.")
            sys.exit()


main()

実行例

oldに256、newに384を入力した例と、oldに384、newに256を入力した例を以下に示します。電卓では手間だけどExcelを使うほどでもないような計算が手軽にできるようになりました。

実行例
c:\work>python myTools.py
== myTools ==
a: comparison of old and new value
x: exit
>a
old>256
new>384
delta:128.000000, times:1.50
== myTools ==
a: comparison of old and new value
x: exit
>a
old>384
new>256
delta:-128.000000, times:0.67
== myTools ==
a: comparison of old and new value
x: exit
>

補足

float型(浮動小数点数)の演算については「15. 浮動小数点演算、その問題と制限(docs.python.org)」もご参考ください。

4. 画像の一部を切り出して保存

Pythonでテストツールを製作する一番のメリットは様々なライブラリの恩恵を受けられることです。そこで、本節ではスクリーンショットの一部を切り出してバグ票に貼り付ける作業を想定し、OpenCVで画像を切り出してファイルに保存する機能を実装します。

要件

  • 以下のパラメータを入力できること
    • 加工元の画像ファイル名
    • 切り出す座標(x0, y0, x1, y1)の値
    • 保存するファイル名
  • 以下の場合はエラーメッセージを表示してメニューに戻ること
    • 加工元ファイルの読み込みに失敗した場合
    • 切り出す座標の値が加工元ファイルの範囲外、x0>x1、y0>y1の場合
    • 切り出した画像の保存に失敗した場合

実装

  • cv2をimportします。
  • パラメータの入力を受け持つcrop_image_menu()と画像を切り出すcrop_image()を実装しmenu()から呼べるようにします。

参考

myTools.py(Ver.3)
# myTools.py by ka's(https://qiita.com/pbjpkas) 2021
# python 3.x
import sys
import cv2


def comparison_value():
    try:
        old_value = float(input("old>"))
        new_value = float(input("new>"))
    except:
        print(sys.exc_info())
        return False

    delta = new_value - old_value
    if old_value != 0:
        times = new_value / old_value
    else:
        times = 0

    print("delta:%f, times:%.2f" % (delta, times))
    return True


def crop_image(filename_in, x0, y0, x1, y1, filename_out):
    img = cv2.imread(filename_in, cv2.IMREAD_COLOR)
    # 加工元ファイルを開けることの確認
    if img is None:
        print("Input File Error.")
        return False

    # 座標がint型にキャストできることを確認
    try:
        v0 = int(y0)
        v1 = int(y1)
        h0 = int(x0)
        h1 = int(x1)
    except:
        print("Coordinate Value Error(1).")
        print(sys.exc_info())
        return False

    # 座標が加工元画像の範囲内、かつ、v0<v1、h0<h1であることの確認
    # print("original image size v:%d, h:%d" % (img.shape[0],img.shape[1]))
    if 0<=v0 and v0<=img.shape[0] and \
       0<=v1 and v1<=img.shape[0] and \
       0<=h0 and h0<=img.shape[1] and \
       0<=h1 and h1<=img.shape[1] and \
       v0<v1 and h0<h1 :
        img2 = img[v0:v1, h0:h1]
    else:
        print("Coordinate Value Error(2).")
        return False

    try:
        ret = cv2.imwrite(filename_out, img2)
        if ret != True:
            print("Output File Error.")
            return False
    except:
        print("Output File Error.")
        print(sys.exc_info())
        return False

    return True


def crop_image_menu():
    filename_in = input("input file>")
    x0 = input("x0>")
    y0 = input("y0>")
    x1 = input("x1>")
    y1 = input("y1>")
    filename_out = input("output file>")
    s = input("OK? y/n>")
    if s == "y":
        ret = crop_image(filename_in, x0, y0, x1, y1, filename_out)
        return ret
    else:
        return False


def main():
    while True:
        print("== myTools ==")
        print("a: comparison of old and new value")
        print("b: crop image")
        print("x: exit")

        s = input(">")

        if s == "a":
            comparison_value()

        if s == "b":
            crop_image_menu()

        if s == "x":
            print("Bye.")
            sys.exit()


main()

実行例

以下のサンプル画像(sample.png1)をC:\workに配置し、赤枠部分(x0:120、y0:38、x1:230、y1:57)を切り出してout.pngに保存します。
sample.png

実行例
c:\work>python myTools.py
== myTools ==
a: comparison of old and new value
b: crop image
x: exit
>b
input file>sample.png
x0>120
y0>38
x1>230
y1>57
output file>out.png
OK? y/n>y
== myTools ==
a: comparison of old and new value
b: crop image
x: exit
>

赤枠部分を切り出すことができました。
out.png

5. おわりに

テストツール製作の一歩目として、基本形のプログラムをカスタマイズしながら数値計算のプログラムと画像を切り出して保存するプログラムを実装しました。Pythonとライブラリを組み合わせることでメニュー形式ならではの操作性、取っつきやすさとライブラリのパワー2を兼ね備えたテストツールを作れます。

筆者の考えるPythonでテストツールを作るメリットを以下に挙げます。

  • ライブラリが豊富
    • UART(RS-232C)通信(PySerial)
    • 計測器の制御(PyVISA)
    • 画像処理(OpenCV、pyocr)
    • シェルコマンドの実行(subprocess)
    • マルチスレッドの処理(threading)
  • たいていのことはググると見つかる
  • Raspberry Piのようなシングルボードコンピュータを治具として利用できる
  • UIをメニュー形式からコマンドライン形式にすることで自動化ツールに流用できる
  • 構造化プログラミング、オブジェクト指向プログラミングのどちらもOK
  • returnで複数のパラメータを返すのが簡単にできる

デメリットとしてPythonやライブラリのバージョン依存問題(はまるとちょっと面倒…)がありますがこれはPythonに限ったことでもないと思います。

Pythonで始めるテストツール製作、おススメです。

付録A. 環境構築

A.1 Pythonのインストール

  1. Python.org → Download
  2. Latest Version(2021/12/4時点では3.10.0)
  3. 一番下の、RecommendedとなっているWindows Installer(64-bit)をダウンロードする python-download.png
  4. Add Python 3.10 to PATHにチェックを入れてからInstall Now 01-python-install-1.png
  5. Disable path length limitを押下してからClose(お好みで) 01-python-install-2.png

A.2 OpenCVのインストール

コマンドプロンプトで以下のpipコマンドを実行します。

pip install opencv-python
pip install opencv-contrib-python

付録B. 動作確認環境

Windows10 64bit バージョン21H1です。Pythonやライブラリのバージョンを以下に示します。

>python --version
Python 3.10.0

>pip list
Package               Version
--------------------- --------
numpy                 1.21.4
opencv-contrib-python 4.5.4.60
opencv-python         4.5.4.60
pip                   21.2.3
setuptools            57.4.0

付録C. 画像切り出しプログラムの動作確認内容

sample.pngは横643ピクセルx縦513ピクセルの画像です。

正常系

1x1ピクセル(切り出しサイズの最小値)、任意のサイズ、全体(切り出しサイズの最大値)の3パターン+αを確認しています。

No. 切り出す箇所 input file x0 y0 x1 y1 output file 期待結果 実行結果
1 赤枠部分 sample.png 120 38 230 57 out.png 赤枠部分がout.pngに保存される OK
2 1x1 pixel sample.png 0 0 1 1 1x1-1.png 元画像左上1x1 pixelが1x1-1.pngに保存される OK
3 1x1 pixel sample.png 642 512 643 513 1x1-2.png 元画像右下1x1 pixelが1x1-2.pngに保存される OK
4 全体 sample.png 0 0 643 513 643x513.png 元画像の全体が643x513.pngに保存される OK
5 左上 sample.png 0 0 20 20 corner-ul.png 元画像左上20x20 pixelがcorner-ul.pngに保存される OK
6 左下 sample.png 0 493 20 513 corner-bl.png 元画像左下20x20 pixelがcorner-bl.pngに保存される OK
7 右上 sample.png 623 0 643 20 corner-ur.png 元画像右上20x20 pixelがcorner-ur.pngに保存される OK
8 右下 sample.png 623 493 643 513 corner-br.png 元画像右下20x20 pixelがcorner-br.pngに保存される OK

エラー処理

Input File Error

No. エラー発生条件 input file x0 y0 x1 y1 output file 期待結果 実行結果
1 存在しないファイルをinput fileに指定する aaa.png 10 10 20 20 bbb.png Input File Error.が返ってくる OK
2 input fileを空欄にする (空欄) 10 10 20 20 bbb.png Input File Error.が返ってくる OK

Output File Error

No. エラー発生条件 input file x0 y0 x1 y1 output file 期待結果 実行結果
1 output fileのパスに存在しないディレクトリを含める sample.png 10 10 20 20 foo\bbb.png Output File Error.が返ってくる OK
2 output fileを空欄にする sample.png 10 10 20 20 (空欄) Output File Error.が返ってくる OK
3 output fileの拡張子を画像以外のものにする sample.png 10 10 20 20 bbb.exe Output File Error.が返ってくる OK

Coordinate Value Error(1)

No. 確認方法 input file x0 y0 x1 y1 output file 期待結果 実行結果
1 切り出す座標の値にアルファベットを与える sample.png a 10 20 20 bbb.png Coordinate Value Error(1).が返ってくる OK
2 切り出す座標の値に小数を含む数値を与える sample.png 10 10 20 20.5 bbb.png Coordinate Value Error(1).が返ってくる OK

Coordinate Value Error(2)

No. 確認方法 input file x0 y0 x1 y1 output file 期待結果 実行結果
1 切り出す座標の値に負の整数を与える sample.png -10 10 20 20 bbb.png Coordinate Value Error(2).が返ってくる OK
2 切り出す座標の値に元画像の範囲外の数値を与える sample.png 10 10 20 514 bbb.png Coordinate Value Error(2).が返ってくる OK
3 切り出す座標の値にx0>x1となる数値を与える sample.png 20 10 10 20 bbb.png Coordinate Value Error(2).が返ってくる OK
4 切り出す座標の値にy0>y1となる数値を与える sample.png 10 20 20 10 bbb.png Coordinate Value Error(2).が返ってくる OK

付録D. ソフトウェアテストのアドベントカレンダー参加記事

  1. 参考：Lチカで始めるテスト自動化(21)キーボード入力待ちの実装 ↩

  2. 例えばsample.pngのタイムスタンプ部分はOpenCVで矩形の白背景や文字列を描画しています。OpenCVを組み込むとWebカメラの映像を動画で撮影したり、フレームを指定して動画から静止画像を切り出したり、画像に図形や文字列を重畳したりといったことも容易にできます。 ↩

1
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
Sign upLogin
1
0