【pytest, 単体テスト】自作classを使ったpytestについてのメモ【fixture】

自作のClassを使ってインスタンスを作成して、pytestを行った時のメモです。

まずはテストしたいプログラム(sample.py)

class SampleClass:
    def get_a(self):
        return "a"

次にテストプログラム(test_sample.py)

import pytest
import sample

@pytest.fixture
def sample_ins():
    return sample.SampleClass() 

def test_sample(sample_ins):
    assert sample_ins.get_a() == "a"
j

後は以下のpytestを実行すると単体テストが走ります。

pytest