How to read argument in diskpart

Question

I have a bat script:

diskpart /s test.txt E:

Diskpart

select disk 2

assign drive letter = %1%

But the output gives a error like, invalid par