Pythonでパスワード付zipファイルでパスワードがかからない

実現したいこと

Pythonでパスワード付zipファイルを作成したい。

発生している問題・分からないこと

Pythonでパスワード付のZIPファイル(複数ファイルを圧縮）作成したいのですが、以下のエラーがでました。
・TypeError: pwd: expected bytes, got str

その後、password =b'1234'　と書き換えたらエラーは出ず、ZIPファイルはできたのですが、肝心のパスワードがかからず、普通にファイルが開いてしまいました。
どの部分が違っているのかわからず困っております。アドバイスよろしくお願いいたします。

該当のソースコード

python3

1import os 2import zipfile 3 4#新しいZIPファイルの名前 5zip_file_name = 'NEW.zip' 6 7#パスワード 8password = b'1234' #←最初はpassword = '1234'で上記のエラーが出たため書き換えました 9 10with zipfile.ZipFile(zip_file_name, 'w', zipfile.ZIP_DEFLATED) as zipf: 11 zipf.setpassword(password) 12 zipf.write('test.txt') 13 zipf.write('test.xlsx') 14 15 16

試したこと・調べたこと

上記の詳細・結果

・TypeError: pwd: expected bytes, got str で調べて　password =b'1234'と書き換えてみました。
・passwordがかからない同様の内容が見つからなかった。

補足

特になし