@@ -11,35 +11,35 @@ class TestTableColumn(unittest.TestCase):
1111
1212 def testWithType (self ):
1313 tc = TableColumn ('foo:int' )
14- assert tc .name () == 'foo'
15- assert tc .type () is int
14+ self . assertEqual ( tc .name (), 'foo' )
15+ self . assertTrue ( tc .type () is int )
1616
1717 def testWithoutType (self ):
1818 tc = TableColumn ('bar' )
19- assert tc .name () == 'bar'
20- assert tc .type () is None
19+ self . assertEqual ( tc .name (), 'bar' )
20+ self . assertTrue ( tc .type () is None )
2121
2222 def testWrongSpec (self ):
2323 self .assertRaises (DataTableError , TableColumn , 'foo:bar' )
2424 self .assertRaises (DataTableError , TableColumn , 'foo:bar:baz' )
2525
2626 def testValueForRawValue (self ):
2727 tc = TableColumn ('foo:int' )
28- assert tc .valueForRawValue ('' ) == 0
29- assert tc .valueForRawValue ('1' ) == 1
30- assert tc .valueForRawValue (2 ) == 2
31- assert tc .valueForRawValue (2.5 ) == 2
28+ self . assertEqual ( tc .valueForRawValue ('' ), 0 )
29+ self . assertEqual ( tc .valueForRawValue ('1' ), 1 )
30+ self . assertEqual ( tc .valueForRawValue (2 ), 2 )
31+ self . assertEqual ( tc .valueForRawValue (2.5 ), 2 )
3232 tc = TableColumn ('bar:str' )
33- assert tc .valueForRawValue ('' ) == ''
34- assert tc .valueForRawValue ('1' ) == '1'
35- assert tc .valueForRawValue (2 ) == '2'
36- assert tc .valueForRawValue ('x' ) == 'x'
33+ self . assertEqual ( tc .valueForRawValue ('' ), '' )
34+ self . assertEqual ( tc .valueForRawValue ('1' ), '1' )
35+ self . assertEqual ( tc .valueForRawValue (2 ), '2' )
36+ self . assertEqual ( tc .valueForRawValue ('x' ), 'x' )
3737 tc = TableColumn ('bar:float' )
38- assert tc .valueForRawValue ('' ) == 0.0
39- assert tc .valueForRawValue ('1' ) == 1.0
40- assert tc .valueForRawValue ('1.5' ) == 1.5
41- assert tc .valueForRawValue (2.5 ) == 2.5
42- assert tc .valueForRawValue (3 ) == 3.0
38+ self . assertEqual ( tc .valueForRawValue ('' ), 0.0 )
39+ self . assertEqual ( tc .valueForRawValue ('1' ), 1.0 )
40+ self . assertEqual ( tc .valueForRawValue ('1.5' ), 1.5 )
41+ self . assertEqual ( tc .valueForRawValue (2.5 ), 2.5 )
42+ self . assertEqual ( tc .valueForRawValue (3 ), 3.0 )
4343
4444
4545class Record (object ):
@@ -61,7 +61,7 @@ def _testSource(self, name, src, headings, data):
6161 dt = DataTable ()
6262 lines = src .splitlines ()
6363 dt .readLines (lines )
64- assert [col .name () for col in dt .headings ()] == headings
64+ self . assertEqual ( [col .name () for col in dt .headings ()], headings )
6565 for i , values in enumerate (dt ):
6666 match = data [i ]
6767 self .assertEqual (values .asList (), match ,
@@ -101,14 +101,15 @@ def _testBasic(self):
101101 t .append (b )
102102 t .append (c )
103103 t .append (d )
104- assert t [- 4 ]['name' ] == 'John'
105- assert t [- 3 ]['name' ] == 'Mary'
106- assert t [- 2 ]['name' ] == 'Fred'
107- assert t [- 1 ]['name' ] == 'Wilma'
108- assert t [- 4 ].asDict () == {'name' : 'John' , 'age' : 26 , 'rating' : 7.25 }
109- assert t [- 3 ].asList () == b
110- assert t [- 2 ].asDict () == c
111- assert t [- 1 ].asList () == ['Wilma' , 27 , 9.5 ]
104+ self .assertEqual (t [- 4 ]['name' ], 'John' )
105+ self .assertEqual (t [- 3 ]['name' ], 'Mary' )
106+ self .assertEqual (t [- 2 ]['name' ], 'Fred' )
107+ self .assertEqual (t [- 1 ]['name' ], 'Wilma' )
108+ self .assertEqual (t [- 4 ].asDict (),
109+ {'name' : 'John' , 'age' : 26 , 'rating' : 7.25 })
110+ self .assertEqual (t [- 3 ].asList (), b )
111+ self .assertEqual (t [- 2 ].asDict (), c )
112+ self .assertEqual (t [- 1 ].asList (), ['Wilma' , 27 , 9.5 ])
112113
113114 # Printing
114115 # print t
@@ -124,13 +125,13 @@ def _testBasic(self):
124125 out = StringIO ()
125126 t .writeFile (out )
126127 results = out .getvalue ()
127- assert results == answer , '\n %r\n %r\n ' % (results , answer )
128+ self . assertEqual ( results , answer , '\n %r\n %r\n ' % (results , answer ) )
128129
129130 # Accessing rows
130131 for row in t :
131- assert row ['name' ] == row [0 ]
132- assert row ['age' ] == row [1 ]
133- assert row ['rating' ] == row [2 ]
132+ self . assertEqual ( row ['name' ], row [0 ])
133+ self . assertEqual ( row ['age' ], row [1 ])
134+ self . assertEqual ( row ['rating' ], row [2 ])
134135 for item in row :
135136 pass
136137
@@ -139,7 +140,7 @@ def _testBasic(self):
139140 t .setHeadings (list ('xyz' ))
140141 t .append ([1 , 2 , 3 ])
141142 t .append ([4 , 5 , 6 ])
142- assert t [0 ]['x' ] - t [1 ]['z' ] == - 5
143+ self . assertEqual ( t [0 ]['x' ] - t [1 ]['z' ], - 5 )
143144
144145 def testBasics (self ):
145146 # Basics
@@ -232,7 +233,7 @@ def _testExcel(self):
232233 # print 'Testing Excel...'
233234 xlsfile = os .path .join (os .path .dirname (__file__ ), 'Sample3.xls' )
234235 t = DataTable (xlsfile )
235- assert t [0 ][0 ] == 1.0 , t [0 ]
236+ self . assertEqual ( t [0 ][0 ], 1.0 , t [0 ])
236237
237238
238239if __name__ == '__main__' :
0 commit comments