pytest使ってみた

インストール

pip install pytest

実装

ディレクトリ構造

今回はこんな感じのディレクトリ構造になっています。

├── main
│   └── calc.py
└── test_calc.py

ソースコード

calc.py

class Calc:
    def __init__(self, a, b):
        self.a = a
        self.b = b
    
    def add(self):
        return self.a + self.b

    def dif(self):
        return self.a - self.b

    def seki(self):
        return self.a * self.b

    def shou(self):
        return self.a / self.b

    

test_calc.py

from main.calc import Calc

def test_add_01():
    assert Calc(7,5).add() == 12

def test_dif_01():
    assert Calc(7,5).dif() == 2

def test_seki_01():
    assert Calc(7,5).seki() == 36

def test_shou_01():
    assert Calc(7,5).shou() == 2

それぞれのソースコードはこんな感じです。
https://qiita.com/kg1/items/4e2cae18e9bd39f014d4
こちらのサイトを参考にさせていただきました。

実行結果

実行はpytestと打つだけです。
以下のような出力がされました。

sekiとshouの部分でわざと間違った計算結果を用いているので正しく動作してそうです。
実際に扱う場面などはまだわからないのでこれから勉強に励んでいきます。