File tree Expand file tree Collapse file tree 1 file changed +45
-0
lines changed Expand file tree Collapse file tree 1 file changed +45
-0
lines changed Original file line number Diff line number Diff line change
1
+ from kagglesdk import kaggle_env
2
+ from kagglesdk import KaggleClient , KaggleEnv
3
+
4
+ # python -m unittest tests.test_authenticate
5
+
6
+ import os
7
+ import unittest
8
+
9
+
10
+ class TestClient (unittest .TestCase ):
11
+
12
+ def setUp (self ):
13
+ print ("setup class:%s" % self )
14
+
15
+ def tearDown (self ):
16
+ print ("teardown class:TestStuff" )
17
+
18
+ # Environment
19
+
20
+ def test_kaggle_environment (self ):
21
+ os .environ ['KAGGLE_API_ENVIRONMENT' ] = 'PROD'
22
+
23
+ env = kaggle_env .get_env ()
24
+ self .assertEqual (env , KaggleEnv .PROD )
25
+
26
+ endpoint = kaggle_env .get_endpoint (env )
27
+ self .assertEqual (endpoint , 'https://www.kaggle.com' )
28
+
29
+ # Client
30
+
31
+ def test_kaggle_client (self ):
32
+ client = KaggleClient (
33
+ env = KaggleEnv .PROD ,
34
+ verbose = False ,
35
+ username = 'dinosaur' ,
36
+ password = 'xxxxxxxxxxxx' )
37
+
38
+ self .assertEqual (client .username , 'dinosaur' )
39
+ self .assertEqual (client .password , 'xxxxxxxxxxxx' )
40
+ self .assertEqual (client .http_client ()._endpoint , 'https://www.kaggle.com' )
41
+ self .assertEqual (client .http_client ()._verbose , False )
42
+
43
+
44
+ if __name__ == '__main__' :
45
+ unittest .main ()
You can’t perform that action at this time.
0 commit comments