def test_set_version_info():
new_version = "100.100.0"
dmp = DagsterModulePublisher()
# Test setting version
version = dmp.set_version_info(new_version=new_version, dry_run=True)
assert version == {"__version__": new_version}
def test_set_version_info():
new_version = '100.100.0'
new_nightly_version = get_nightly_version()
dmp = DagsterModulePublisher()
# Test setting both version and nightly
version = dmp.set_version_info(new_version=new_version, dry_run=True)
assert version == {
'__version__': new_version,
'__nightly__': new_nightly_version
}
# Test only setting nightly
version = dmp.set_version_info(dry_run=True)
dm = DagsterModule('dagster', is_library=False)
existing_version = dm.get_version_info()['__version__']
assert version == {
'__version__': existing_version,
'__nightly__': new_nightly_version
}