Updated tests

Author: akshowhini

tests/constants.py

@@ -1,3 +1,2 @@
 API_KEY = ''
 FILE_PATH = r''
-RESULTS_FOLDER = r''

tests/test_client.py

@@ -1,37 +1,35 @@
+import os
 import io
 
 import pytest
 
 from ExtractTable.client import ExtractTable
-from ExtractTable.common import UsageStats
 from ExtractTable.exceptions import ServiceError
-from tests.constants import API_KEY, FILE_PATH, RESULTS_FOLDER
+from tests.constants import API_KEY, FILE_PATH
 
 
 @pytest.fixture
 def client():
     return ExtractTable(API_KEY)
 
 
-def test_process_file(client: ExtractTable):
-    assert not (client.process_file(FILE_PATH, RESULTS_FOLDER))
-
-
-def test_process_file_index(client: ExtractTable):
-    assert not (client.process_file(FILE_PATH, RESULTS_FOLDER, True))
-
-
-def test_check_usage(client: ExtractTable):
-    assert isinstance(client.check_usage(), UsageStats)
+def test_get_result_fail(client: ExtractTable):
+    with pytest.raises(ServiceError):
+        assert not client.get_result('')
 
 
 def test_trigger_process_fail(client: ExtractTable):
     with pytest.raises(ServiceError):
         client.trigger_process(io.BytesIO())
 
 
-def test_get_result_fail(client: ExtractTable):
-    with pytest.raises(ServiceError):
-        assert client.get_result('')
+def test_check_usage(client: ExtractTable):
+    assert isinstance(client.check_usage(), dict)
+
+
+def test_process_file(client: ExtractTable):
+    assert isinstance(client.process_file(filepath=FILE_PATH, output_format="df"), list)
 
 
+def test_process_file_csv(client: ExtractTable, fmt="csv"):
+    return all([os.path.exists(x) and x.endswith(fmt) for x in client.process_file(filepath=FILE_PATH, output_format=fmt)])