import pytest from aircox.models import Station, Program class TestProgramQuerySet: @pytest.mark.django_db def test_station(self, stations, programs): prog = Program.objects.station(station=stations[0]) assert programs[0] in prog assert programs[2] not in prog class TestProgram: @pytest.mark.django_db def test_station(self, stations): pass