import os
from importlib import find_loader
from importlib.util import find_spec

folder = "test.tmp"
modulename1 = "spam"
modulename2 = "more_spam"

with open(modulename1+".py", "w") as f:
	f.write("foo = 1")

if not os.path.isdir(folder):
	os.mkdir(folder)
with open(os.path.join(folder,modulename2+".py"), "w") as f:
	f.write("foo = 1")

assert(find_loader(modulename1,".") != None)
assert(find_spec(modulename1,".") != None)

assert(find_loader(modulename1, folder) == None) # fails
assert(find_spec(modulename1, folder) == None) # fails

assert(find_loader(modulename2, folder) != None) # fails
assert(find_spec(modulename2, folder) != None) # fails

os.chdir(folder)

assert(find_loader(modulename1,".") == None)
assert(find_spec(modulename1,".") == None) # fails

assert(find_loader(modulename2,".") != None)
assert(find_spec(modulename2,".") != None) # fails

os.chdir("..")