@@ -11,35 +11,35 @@ class TestTableColumn(unittest.TestCase):
11
11
12
12
def testWithType (self ):
13
13
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 )
16
16
17
17
def testWithoutType (self ):
18
18
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 )
21
21
22
22
def testWrongSpec (self ):
23
23
self .assertRaises (DataTableError , TableColumn , 'foo:bar' )
24
24
self .assertRaises (DataTableError , TableColumn , 'foo:bar:baz' )
25
25
26
26
def testValueForRawValue (self ):
27
27
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 )
32
32
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' )
37
37
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 )
43
43
44
44
45
45
class Record (object ):
@@ -61,7 +61,7 @@ def _testSource(self, name, src, headings, data):
61
61
dt = DataTable ()
62
62
lines = src .splitlines ()
63
63
dt .readLines (lines )
64
- assert [col .name () for col in dt .headings ()] == headings
64
+ self . assertEqual ( [col .name () for col in dt .headings ()], headings )
65
65
for i , values in enumerate (dt ):
66
66
match = data [i ]
67
67
self .assertEqual (values .asList (), match ,
@@ -101,14 +101,15 @@ def _testBasic(self):
101
101
t .append (b )
102
102
t .append (c )
103
103
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 ])
112
113
113
114
# Printing
114
115
# print t
@@ -124,13 +125,13 @@ def _testBasic(self):
124
125
out = StringIO ()
125
126
t .writeFile (out )
126
127
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 ) )
128
129
129
130
# Accessing rows
130
131
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 ])
134
135
for item in row :
135
136
pass
136
137
@@ -139,7 +140,7 @@ def _testBasic(self):
139
140
t .setHeadings (list ('xyz' ))
140
141
t .append ([1 , 2 , 3 ])
141
142
t .append ([4 , 5 , 6 ])
142
- assert t [0 ]['x' ] - t [1 ]['z' ] == - 5
143
+ self . assertEqual ( t [0 ]['x' ] - t [1 ]['z' ], - 5 )
143
144
144
145
def testBasics (self ):
145
146
# Basics
@@ -232,7 +233,7 @@ def _testExcel(self):
232
233
# print 'Testing Excel...'
233
234
xlsfile = os .path .join (os .path .dirname (__file__ ), 'Sample3.xls' )
234
235
t = DataTable (xlsfile )
235
- assert t [0 ][0 ] == 1.0 , t [0 ]
236
+ self . assertEqual ( t [0 ][0 ], 1.0 , t [0 ])
236
237
237
238
238
239
if __name__ == '__main__' :
0 commit comments