Skip to content

Commit 9a8e456

Browse files
authored
Add simple unit test to kagglesdk. (#671)
Test initialization of KaggleEnv and KaggleClient, based on build_kaggle_client() in kaggle/api/kaggle_api_extended.py.
1 parent fa95b9c commit 9a8e456

File tree

1 file changed

+45
-0
lines changed

1 file changed

+45
-0
lines changed

kagglesdk/test/test_client.py

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
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()

0 commit comments

Comments
 (0)