tests/test_mb_moe.py

import megablocks

def test_import():
    """Simple test to check if the module can be imported."""
    print("megablocks_moe module imported successfully.")
    print("Available functions:", dir(megablocks))