DeMemWM / tests /test_dememwm_no_old_path_mutation_static.py
BonanDing's picture
Initial commit
b47a1ce
raw
history blame contribute delete
879 Bytes
from pathlib import Path
def test_new_standalone_files_do_not_reference_old_ssm_method_paths():
forbidden = [
"spatial_ssm_memory",
"df_video_ssm_memory",
"StateSpaceSpatialMemoryMinecraft",
"ssm_memory_ckpt_path",
"dememwm_compression",
]
for rel in ["algorithms/worldmem/dememwm_memory_dit.py", "algorithms/worldmem/dememwm/algorithm.py", "configurations/algorithm/dememwm_memory_dit.yaml"]:
text = Path(rel).read_text()
for token in forbidden:
assert token not in text, f"{token} leaked into {rel}"
def test_legacy_ssm_bridge_scaffold_is_not_present():
assert not Path("algorithms/worldmem/models/dememwm_compression.py").exists()
assert not Path("algorithms/worldmem/df_video_ssm_memory.py").exists()
assert not Path("algorithms/worldmem/models/spatial_ssm_memory.py").exists()