Changed the get_memory check for number.
This commit is contained in:
parent
9f12616eca
commit
0a956e8500
6
main.py
6
main.py
|
@ -5,7 +5,7 @@ def get_memory(variable_name: str) -> list[bool]:
|
|||
while True:
|
||||
input_chars: list[str] = list(input(f"Enter {variable_name}: "))
|
||||
|
||||
if all(character.isnumeric() and (character == "1" or character == "0") for character in input_chars):
|
||||
if all(character in ["0", "1"] for character in input_chars):
|
||||
return [True if character == "1" else False for character in input_chars]
|
||||
else:
|
||||
print(f"[ERROR] The {variable_name} may contain only 1-s and 0-s!")
|
||||
|
@ -16,8 +16,8 @@ if __name__ == '__main__':
|
|||
|
||||
print(reg)
|
||||
|
||||
print(reg.left_shift(3))
|
||||
print(reg.left_shift())
|
||||
print(reg)
|
||||
|
||||
print(reg.right_shift(3))
|
||||
print(reg.right_shift())
|
||||
print(reg)
|
||||
|
|
Loading…
Reference in New Issue