mirror of
https://github.com/RustPython/RustPython.git
synced 2026-10-29 11:14:38 +08:00
50 lines
982 B
Python
50 lines
982 B
Python
from testutils import assert_raises
|
|
|
|
import csv
|
|
|
|
for row in csv.reader(["one,two,three"]):
|
|
[one, two, three] = row
|
|
assert one == "one"
|
|
assert two == "two"
|
|
assert three == "three"
|
|
|
|
|
|
def f():
|
|
iter = ["one,two,three", "four,five,six"]
|
|
reader = csv.reader(iter)
|
|
|
|
[one, two, three] = next(reader)
|
|
[four, five, six] = next(reader)
|
|
|
|
assert one == "one"
|
|
assert two == "two"
|
|
assert three == "three"
|
|
assert four == "four"
|
|
assert five == "five"
|
|
assert six == "six"
|
|
|
|
|
|
f()
|
|
|
|
|
|
def test_delim():
|
|
iter = ["one|two|three", "four|five|six"]
|
|
reader = csv.reader(iter, delimiter="|")
|
|
|
|
[one, two, three] = next(reader)
|
|
[four, five, six] = next(reader)
|
|
|
|
assert one == "one"
|
|
assert two == "two"
|
|
assert three == "three"
|
|
assert four == "four"
|
|
assert five == "five"
|
|
assert six == "six"
|
|
|
|
with assert_raises(TypeError):
|
|
iter = ["one,,two,,three"]
|
|
csv.reader(iter, delimiter=",,")
|
|
|
|
|
|
test_delim()
|