From b2a6afa68ec49734659ceeb7961218cf5f17b204 Mon Sep 17 00:00:00 2001 From: Sylvain Date: Wed, 4 Mar 2009 11:54:23 +0100 Subject: toward py3k passing tests: unittest_shellutils, unittest_textutils, unittest_modutils, unittest_table, unittest_tree --- test/unittest_table.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'test/unittest_table.py') diff --git a/test/unittest_table.py b/test/unittest_table.py index d41c646..cd2e9a5 100644 --- a/test/unittest_table.py +++ b/test/unittest_table.py @@ -2,11 +2,12 @@ Unittests for table management """ -__revision__ = '$Id: unittest_table.py,v 1.13 2006-04-09 22:30:53 nico Exp $' - import sys import os -from cStringIO import StringIO +try: + from cStringIO import StringIO +except ImportError: + from io import StringIO from logilab.common.testlib import TestCase, unittest_main from logilab.common.table import Table, TableStyleSheet, DocbookTableWriter, \ @@ -55,8 +56,8 @@ class TableTC(TestCase): def test_iterable(self): """test iter(table)""" it = iter(self.table) - self.assert_(it.next() == self.table.data[0]) - self.assert_(it.next() == self.table.data[1]) + self.assert_(next(it) == self.table.data[0]) + self.assert_(next(it) == self.table.data[1]) def test_get_rows(self): """tests Table.get_rows()""" -- cgit v1.2.1