9
9
import math
10
10
import os
11
11
import sys
12
- import six
13
12
import tempfile
14
13
from builtins import zip
15
14
from mock import MagicMock
16
- from nose .tools import assert_raises , assert_equals , assert_not_equals , raises , assert_false , assert_not_in , assert_sequence_equal , assert_in
15
+ from nose .tools import assert_raises , assert_equals , assert_not_equals , raises , assert_false , assert_not_in , assert_sequence_equal
17
16
from nose import SkipTest
18
17
import unit
19
18
23
22
except ImportError :
24
23
pandas_found = True
25
24
26
- from nose .tools import raises , assert_equals , assert_is_instance
25
+ from nose .tools import raises , assert_equals , assert_set_equal
27
26
import unit
28
27
import synapseclient
29
28
from synapseclient import Entity
@@ -314,15 +313,13 @@ def test_csv_table():
314
313
# print(table_row, expected_row)
315
314
assert table_row == expected_row
316
315
317
- expected_rows = [[str (val ) for val in row ] for row in data ]
318
-
319
316
## test asRowSet
320
317
rowset = table .asRowSet ()
321
- for rowset_row , expected_row in zip (rowset .rows , expected_rows ):
318
+ for rowset_row , expected_row in zip (rowset .rows , data ):
322
319
#print(rowset_row, expected_row)
323
- assert_equals ( rowset_row ['values' ], expected_row [2 :])
324
- assert_equals ( rowset_row ['rowId' ], expected_row [0 ])
325
- assert_equals ( rowset_row ['versionNumber' ], expected_row [1 ])
320
+ assert rowset_row ['values' ]== expected_row [2 :]
321
+ assert rowset_row ['rowId' ]== expected_row [0 ]
322
+ assert rowset_row ['versionNumber' ]== expected_row [1 ]
326
323
327
324
## test asDataFrame
328
325
try :
@@ -371,15 +368,12 @@ def test_list_of_rows_table():
371
368
## need columns to do cast_values w/o storing
372
369
table = Table (schema1 , data , headers = [SelectColumn .from_column (col ) for col in cols ])
373
370
374
-
375
371
for table_row , expected_row in zip (table , data ):
376
- assert_equals (table_row , expected_row )
377
-
378
- expected_rows = [[str (val ) for val in row ] for row in data ]
372
+ assert table_row == expected_row
379
373
380
374
rowset = table .asRowSet ()
381
- for rowset_row , expected_row in zip (rowset .rows , expected_rows ):
382
- assert_equals ( rowset_row ['values' ], expected_row )
375
+ for rowset_row , expected_row in zip (rowset .rows , data ):
376
+ assert rowset_row ['values' ]== expected_row
383
377
384
378
table .columns = cols
385
379
@@ -718,22 +712,6 @@ def test_constructor__name_to_col_id(self):
718
712
assert_equals (711 , partial_row .rowId )
719
713
720
714
721
- def test_values_have_string_type (type ):
722
- values = {
723
- '12three' :321 ,
724
- 456 :'65four'
725
- }
726
- partial_row = PartialRow (values , rowId = 11111 )
727
- for key_val in partial_row .values :
728
- assert_is_instance (key_val ['key' ], six .string_types )
729
- assert_is_instance (key_val ['value' ], six .string_types )
730
-
731
- expected_values = [{'key' :'12three' , 'value' :'321' }, {'key' :'456' , 'value' :'65four' }]
732
- assert_equals (2 , len (expected_values ))
733
- assert_in (expected_values [0 ], partial_row .values )
734
- assert_in (expected_values [1 ], partial_row .values )
735
-
736
-
737
715
class TestPartialRowSet ():
738
716
@raises (ValueError )
739
717
def test_constructor__not_all_rows_of_type_PartialRow (self ):
@@ -747,9 +725,3 @@ def test_constructor__single_PartialRow(self):
747
725
assert_equals ([partial_row ], partial_rowset .rows )
748
726
749
727
750
- class TestRow ():
751
- def test_values_have_string_type (self ):
752
- row = Row ([1 ,2 ,"three" ])
753
- for val in row .values :
754
- assert_is_instance (val , six .string_types )
755
- assert_equals (["1" ,"2" ,"three" ], row .values )
0 commit comments