def test_get_models_from_path(self):
# test with no models
path = Path(__file__)
models = em.get_models_from_path(path)
assert len(models) == 0
# test a directory
path = path.parent / 'dummy_models'
models = em.get_models_from_path(path)
assert len(models) == 2
# test a file
path = path / 'dummies.py'
models = em.get_models_from_path(path)
assert len(models) == 2
def test_dont_get_models_from_dunders(self):
# test a path that has a dunder fails
assert 0 == len(em.get_models_from_path(Path('foo/__bar__.py')))
assert 0 == len(em.get_models_from_path(Path('foo/__pycache')))
def test_dont_get_models_from_non_py_files(self):
assert 0 == len(em.get_models_from_path(Path('foo.py.swp')))