33from unittest import TestCase
44import pyexcel as pe
55import pyexcel_webio as webio
6+ from common import TestInput , TestExtendedInput
67from db import Session , Base , Signature , Signature2 , engine
78from nose .tools import raises , eq_
89if sys .version_info [0 ] == 2 and sys .version_info [1 ] < 7 :
1415OUTPUT = "%s.xls" % FILE_NAME
1516
1617
17- class TestInput (webio .ExcelInput ):
18- """This is sample implementation that read excel source from file"""
19- def get_params (self , ** keywords ):
20- """Load a single sheet"""
21- return keywords
22-
23-
24- class TestExtendedInput (webio .ExcelInputInMultiDict ):
25- """This is sample implementation that read excel source from file"""
26- def get_file_tuple (self , field_name ):
27- return field_name
28-
29-
3018def dumpy_response (content , content_type = None , status = 200 , file_name = None ):
3119 """A dummy response"""
3220 with open (file_name , 'wb' ) as f :
@@ -52,32 +40,32 @@ def test_excel_input_get_file_tuple(self):
5240 testinput = webio .ExcelInputInMultiDict ()
5341 testinput .get_file_tuple (field_name = "test" ) # booom
5442
55- @raises (NotImplementedError )
43+ @raises (pe . sources . factory . UnknownParameters )
5644 def test_get_sheet (self ):
5745 myinput = TestInput ()
5846 myinput .get_sheet (unrelated = "foo bar" )
5947
60- @raises (NotImplementedError )
48+ @raises (pe . sources . factory . UnknownParameters )
6149 def test_get_array (self ):
6250 myinput = TestInput ()
6351 myinput .get_array (unrelated = "foo bar" )
6452
65- @raises (NotImplementedError )
53+ @raises (pe . sources . factory . UnknownParameters )
6654 def test_get_dict (self ):
6755 myinput = TestInput ()
6856 myinput .get_dict (unrelated = "foo bar" )
6957
70- @raises (NotImplementedError )
58+ @raises (pe . sources . factory . UnknownParameters )
7159 def test_get_records (self ):
7260 myinput = TestInput ()
7361 myinput .get_records (unrelated = "foo bar" )
7462
75- @raises (NotImplementedError )
63+ @raises (pe . sources . factory . UnknownParameters )
7664 def test_get_book (self ):
7765 myinput = TestInput ()
7866 myinput .get_book (unrelated = "foo bar" )
7967
80- @raises (NotImplementedError )
68+ @raises (pe . sources . factory . UnknownParameters )
8169 def test_get_book_dict (self ):
8270 myinput = TestInput ()
8371 myinput .get_book_dict (unrelated = "foo bar" )
0 commit comments