Add more test for csv

libkluid · libkluid · commit 25f656105f8e · 2019-10-13T17:20:33.000+09:00
diff --git a/tests/snippets/test_csv.py b/tests/snippets/test_csv.py
@@ -1,3 +1,5 @@
+from testutils import assert_raises
+
 import csv
 
 for row in csv.reader(['one,two,three']):
@@ -21,3 +23,23 @@ def f():
 	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()
