format

airbytehq · Jul 12, 2021 · 323ff30 · 323ff30
1 parent 36a18d4
commit 323ff30
Showing 1 changed file with 12 additions and 31 deletions.
diff --git a/airbyte-integrations/connectors/source-dixa/unit_tests/unit_test.py b/airbyte-integrations/connectors/source-dixa/unit_tests/unit_test.py
@@ -20,6 +20,8 @@
 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
+#
+
 from datetime import datetime, timezone
 
 import pytest
@@ -28,9 +30,7 @@
 
 @pytest.fixture
 def conversation_export():
-    return ConversationExport(
-        start_date=datetime(year=2021, month=7, day=1, hour=12, tzinfo=timezone.utc), batch_size=1, logger=None
-    )
+    return ConversationExport(start_date=datetime(year=2021, month=7, day=1, hour=12, tzinfo=timezone.utc), batch_size=1, logger=None)
 
 
 def test_validate_ms_timestamp_with_valid_input():
@@ -67,14 +67,8 @@ def test_add_days_to_ms_timestamp():
 def test_stream_slices_without_state(conversation_export):
     conversation_export.end_timestamp = 1625270400001  # 2021-07-03 00:00:00 + 1 ms
     expected_slices = [
-        {
-            'updated_after': 1625140800000,  # 2021-07-01 12:00:00
-            'updated_before': 1625227200000  # 2021-07-02 12:00:00
-        },
-        {
-            'updated_after': 1625227200000,
-            'updated_before': 1625270400001
-        }
+        {"updated_after": 1625140800000, "updated_before": 1625227200000},  # 2021-07-01 12:00:00  # 2021-07-02 12:00:00
+        {"updated_after": 1625227200000, "updated_before": 1625270400001},
     ]
     actual_slices = conversation_export.stream_slices()
     assert actual_slices == expected_slices
@@ -85,12 +79,7 @@ def test_stream_slices_without_state_large_batch():
         start_date=datetime(year=2021, month=7, day=1, hour=12, tzinfo=timezone.utc), batch_size=31, logger=None
     )
     conversation_export.end_timestamp = 1625270400001  # 2021-07-03 00:00:00 + 1 ms
-    expected_slices = [
-        {
-            'updated_after': 1625140800000,  # 2021-07-01 12:00:00
-            'updated_before': 1625270400001
-        }
-    ]
+    expected_slices = [{"updated_after": 1625140800000, "updated_before": 1625270400001}]  # 2021-07-01 12:00:00
     actual_slices = conversation_export.stream_slices()
     assert actual_slices == expected_slices
 
@@ -107,26 +96,18 @@ def test_stream_slices_with_start_timestamp_larger_than_state():
     Test that if start_timestamp is larger than state, then start at start_timestamp.
     """
     conversation_export = ConversationExport(
-        start_date=datetime(year=2021, month=12, day=1, tzinfo=timezone.utc), batch_size=31,
-        logger=None
+        start_date=datetime(year=2021, month=12, day=1, tzinfo=timezone.utc), batch_size=31, logger=None
     )
     conversation_export.end_timestamp = 1638360000001  # 2021-12-01 12:00:00 + 1 ms
-    expected_slices = [
-        {
-            'updated_after': 1638316800000,  # 2021-07-01 12:00:00
-            'updated_before': 1638360000001
-        }
-    ]
-    actual_slices = conversation_export.stream_slices(
-        stream_state={'updated_at': 1625220000000}  # # 2021-07-02 12:00:00
-    )
+    expected_slices = [{"updated_after": 1638316800000, "updated_before": 1638360000001}]  # 2021-07-01 12:00:00
+    actual_slices = conversation_export.stream_slices(stream_state={"updated_at": 1625220000000})  # # 2021-07-02 12:00:00
     assert actual_slices == expected_slices
 
 
 def test_get_updated_state_without_state(conversation_export):
-    assert conversation_export.get_updated_state(
-        current_stream_state=None, latest_record={'updated_at': 1625263200000}
-    ) == {'updated_at': 1625140800000}
+    assert conversation_export.get_updated_state(current_stream_state=None, latest_record={"updated_at": 1625263200000}) == {
+        "updated_at": 1625140800000
+    }
 
 
 def test_get_updated_state_with_bigger_state(conversation_export):