dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed error message in debase

Browse Source
This commit is contained in:
Benoît Sierro
2023-12-13 10:56:06 +01:00
parent 1f60aaa65c
commit 2d1cf0c9c1
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/scgenerator/variableparameters.py
View File

@@ -268,12 +268,13 @@ def debase(num: int, base: tuple[int, ...], strict: bool = True) -> tuple[int, .
``` ```
""" """
indices = [] indices = []
orig_num = num
for i in range(len(base) - 1): for i in range(len(base) - 1):
ind, num = divmod(num, np.prod(base[i + 1 :])) ind, num = divmod(num, np.prod(base[i + 1 :]))
indices.append(ind) indices.append(ind)
indices.append(num) indices.append(num)
if strict and indices[0] >= base[0]: if strict and indices[0] >= base[0]:
raise ValueError(f"{num} too big for base {base!r} in strict mode") raise ValueError(f"{orig_num} too big for base {base!r} in strict mode")
return tuple(indices) return tuple(indices)

2
tests/test_variableparameters.py
View File

@@ -19,6 +19,8 @@ def test_debase():
debase(785, (5, 1, 6)) debase(785, (5, 1, 6))
assert debase(785, (5, 1, 6), strict=False) == (130, 0, 5) assert debase(785, (5, 1, 6), strict=False) == (130, 0, 5)
assert debase(0, (1, 1, 1), strict=True) == (0, 0, 0)
def test_get_sequence(): def test_get_sequence():
s = get_sequence([[0, 0, 20]]) s = get_sequence([[0, 0, 20]])