def load_version(version_file: str = VERSION_FILE) -> PlenumVersion:
with open(version_file, 'r') as _f:
version = json.load(_f)
if not isinstance(version, collections.abc.Iterable):
raise InvalidVersionError(
"Failed to load from {}: data '{}' is not iterable".format(
version_file, version))
return PlenumVersion('.'.join([str(i) for i in version if str(i)]))
def set_version(version: str, version_file: str = VERSION_FILE):
version = PlenumVersion(version)
with open(version_file, 'w') as _f:
json.dump(['' if i is None else i for i in version.parts], _f)
_f.write('\n')
def test_plenum_version_upstrean():
pv = PlenumVersion('1.2.3')
assert pv.upstream is pv
def test_plenum_version_parts():
assert PlenumVersion('1.2.3.dev2').parts == (1, 2, 3, 'dev', 2)
assert PlenumVersion('1.2.3.rc3').parts == (1, 2, 3, 'rc', 3)
assert PlenumVersion('1.2.3').parts == (1, 2, 3, None, None)
def test_plenum_version_valid(version):
PlenumVersion(version)
def test_plenum_version_invalid_value(version):
with pytest.raises(InvalidVersionError):
PlenumVersion(version)
def test_set_load_version_valid(version, version_file_path):
set_version(version, version_file_path)
assert load_version(version_file_path) == PlenumVersion(version)
