# Path: pywr/core.py



import os
import pywr.core
import pandas
from numpy.testing import assert_allclose
from pywr.core import Model, Timestep


def load_model(filename=None, data=None, solver=None, model_klass=None, check=True):
    """Load a test model and check it"""
    if data is None:
        path = os.path.join(os.path.dirname(__file__), "models")
        with open(os.path.join(path, filename), "r") as f:
            data = f.read()
    else:
        path = None

    if model_klass is None:

# Path: pywr/core.py



# Path: tests/helpers.py
import os
import pywr.core
import pandas
from numpy.testing import assert_allclose
from pywr.core import Model, Timestep



def load_model(filename=None, data=None, solver=None, model_klass=None, check=True):
    """Load a test model and check it"""
    if data is None:
        path = os.path.join(os.path.dirname(__file__), "models")
        with open(os.path.join(path, filename), "r") as f:
            data = f.read()
    else:
        path = None

    if model_klass is None:

        model_klass = Model