From 2b8456ba346ac045b9482bf52354aa1170ce3adf Mon Sep 17 00:00:00 2001 From: Paul Makepeace Date: Wed, 20 Jul 2011 15:31:47 +0000 Subject: [PATCH] Read whole file for completeness. Include an example of how to use csv.reader to parse the incoming export --- tests/test_refine.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/test_refine.py b/tests/test_refine.py index ad5834f..d1cd4a0 100644 --- a/tests/test_refine.py +++ b/tests/test_refine.py @@ -9,6 +9,7 @@ GOOGLE_REFINE_HOST and GOOGLE_REFINE_PORT. # Copyright (c) 2011 Paul Makepeace, Real Programmers. All rights reserved. +import csv import unittest from google.refine import refine @@ -61,6 +62,17 @@ class RefineTest(refinetest.RefineTestCase): fp = refine.RefineProject(self.project.project_url()).export() line = fp.next() self.assertTrue('email' in line) + for line in fp: + self.assertTrue('M' in line or 'F' in line) + fp.close() + + def test_open_export_csv(self): + fp = refine.RefineProject(self.project.project_url()).export() + csv_fp = csv.reader(fp, dialect='excel-tab') + row = csv_fp.next() + self.assertTrue(row[0] == 'email') + for row in csv_fp: + self.assertTrue(row[3] == 'F' or row[3] == 'M') fp.close()