@@ -121,37 +121,37 @@ def test_datetime_formats(self):
121121 self .assertEqual (tbl [0 ][0 ], expected )
122122
123123 def test_merge_allnull (self ):
124- t1 = agate_helper . table_from_rows ([(1 , "a" , None ), (2 , "b" , None )], ("a" , "b" , "c" ))
125- t2 = agate_helper . table_from_rows ([(3 , "c" , None ), (4 , "d" , None )], ("a" , "b" , "c" ))
124+ t1 = agate . Table ([(1 , "a" , None ), (2 , "b" , None )], ("a" , "b" , "c" ))
125+ t2 = agate . Table ([(3 , "c" , None ), (4 , "d" , None )], ("a" , "b" , "c" ))
126126 result = agate_helper .merge_tables ([t1 , t2 ])
127127 self .assertEqual (result .column_names , ("a" , "b" , "c" ))
128- assert isinstance (result .column_types [0 ], agate_helper . Integer )
128+ assert isinstance (result .column_types [0 ], agate . data_types . Number )
129129 assert isinstance (result .column_types [1 ], agate .data_types .Text )
130130 assert isinstance (result .column_types [2 ], agate .data_types .Number )
131131 self .assertEqual (len (result ), 4 )
132132
133133 def test_merge_mixed (self ):
134- t1 = agate_helper . table_from_rows ([(1 , "a" , None ), (2 , "b" , None )], ("a" , "b" , "c" ))
135- t2 = agate_helper . table_from_rows ([(3 , "c" , "dog" ), (4 , "d" , "cat" )], ("a" , "b" , "c" ))
136- t3 = agate_helper . table_from_rows ([(3 , "c" , None ), (4 , "d" , None )], ("a" , "b" , "c" ))
134+ t1 = agate . Table ([(1 , "a" , None ), (2 , "b" , None )], ("a" , "b" , "c" ))
135+ t2 = agate . Table ([(3 , "c" , "dog" ), (4 , "d" , "cat" )], ("a" , "b" , "c" ))
136+ t3 = agate . Table ([(3 , "c" , None ), (4 , "d" , None )], ("a" , "b" , "c" ))
137137
138138 result = agate_helper .merge_tables ([t1 , t2 ])
139139 self .assertEqual (result .column_names , ("a" , "b" , "c" ))
140- assert isinstance (result .column_types [0 ], agate_helper . Integer )
140+ assert isinstance (result .column_types [0 ], agate . data_types . Number )
141141 assert isinstance (result .column_types [1 ], agate .data_types .Text )
142142 assert isinstance (result .column_types [2 ], agate .data_types .Text )
143143 self .assertEqual (len (result ), 4 )
144144
145145 result = agate_helper .merge_tables ([t2 , t3 ])
146146 self .assertEqual (result .column_names , ("a" , "b" , "c" ))
147- assert isinstance (result .column_types [0 ], agate_helper . Integer )
147+ assert isinstance (result .column_types [0 ], agate . data_types . Number )
148148 assert isinstance (result .column_types [1 ], agate .data_types .Text )
149149 assert isinstance (result .column_types [2 ], agate .data_types .Text )
150150 self .assertEqual (len (result ), 4 )
151151
152152 result = agate_helper .merge_tables ([t1 , t2 , t3 ])
153153 self .assertEqual (result .column_names , ("a" , "b" , "c" ))
154- assert isinstance (result .column_types [0 ], agate_helper . Integer )
154+ assert isinstance (result .column_types [0 ], agate . data_types . Number )
155155 assert isinstance (result .column_types [1 ], agate .data_types .Text )
156156 assert isinstance (result .column_types [2 ], agate .data_types .Text )
157157 self .assertEqual (len (result ), 6 )
@@ -191,7 +191,7 @@ def test_nocast_bool_01(self):
191191 self .assertEqual (len (tbl ), len (result_set ))
192192
193193 assert isinstance (tbl .column_types [0 ], agate .data_types .Boolean )
194- assert isinstance (tbl .column_types [1 ], agate_helper . Integer )
194+ assert isinstance (tbl .column_types [1 ], agate . data_types . Number )
195195
196196 expected = [
197197 [True , Decimal (1 )],
0 commit comments