from src.style_diffuser import StyleDiffuser
assert StyleDiffuser().diffuse(['x'],4)[0].startswith('    ')