test_zone.py

import pandas as pd
from minisim import Infiltration, Inhabitant, Zone, Weather

import pytest

# using an instance to pass global-like variables among tests
class Vars: pass


@pytest.fixture
def basic_variables():
    infiltration = Infiltration("infiltration", 5)
    bob = Inhabitant('Bob', 36000)
    mary = Inhabitant ('Mary', 36000)
    dilbert = Inhabitant('Dilbert', 4000)
    weather1 = Weather(400)
    weather2 = Weather(800)
    hourly_2hours = pd.date_range('1/1/2017', periods = 2, freq = 'H')
    hourly_2days  = pd.date_range('1/1/2017', periods = 2*24, freq = 'H')
    hourly_2weeks = pd.date_range('1/1/2017', periods = 2*7*24, freq = 'H')
    vars = Vars()
    vars.__dict__ = locals()
    return vars
    

def test_zone_basic(basic_variables):
    vars = basic_variables
    infiltration = vars.infiltration
    bob          = vars.bob
    mary         = vars.mary
    weather      = vars.weather
    date_range   = vars.hourly_2hours

    zone = Zone(50, infiltration, inhabitants = [bob])
    zone.add_inhabitants(mary)
    simulation = zone.simulate(date_range, weather, 400)
    print(simulation)

test_zone2.py

import pandas as pd
from minisim import Infiltration, Inhabitant, Zone, Weather

import pytest

@pytest.fixture
def basic_variables():
    infiltration = Infiltration("infiltration", 5)
    bob = Inhabitant('Bob', 36000)
    mary = Inhabitant ('Mary', 36000)
    dilbert = Inhabitant('Dilbert', 4000)
    weather1 = Weather(400)
    weather2 = Weather(800)
    hourly_2hours = pd.date_range('1/1/2017', periods = 2, freq = 'H')
    hourly_2days  = pd.date_range('1/1/2017', periods = 2*24, freq = 'H')
    hourly_2weeks = pd.date_range('1/1/2017', periods = 2*7*24, freq = 'H')
    return locals()


def test_zone_basic(basic_variables):
    vars = basic_variables
    infiltration, bob, mary, weather, hourly_2weeks = **vars

    zone = Zone(50, infiltration, inhabitants = [bob])
    zone.add_inhabitants(mary)
    simulation = zone.simulate(date_range, weather, 400)
    print(simulation)