Skip to content

Commit

Permalink
black format
Browse files Browse the repository at this point in the history
  • Loading branch information
Torgayev Tamirlan committed Mar 9, 2020
1 parent 7ea139c commit e24dff1
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions tests/test_kinesis.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,13 +8,13 @@ def test_create_record(self):
data = "test_data"
record = kinesis.create_record(data)

self.assertTrue('Data' in record)
self.assertTrue('PartitionKey' in record)
self.assertTrue("Data" in record)
self.assertTrue("PartitionKey" in record)

self.assertEqual(record['Data'], b'test_data')
self.assertEqual(record["Data"], b"test_data")

self.assertTrue(type(record['PartitionKey']) == str)
self.assertTrue(len(record['PartitionKey']) <= 256)
self.assertTrue(type(record["PartitionKey"]) == str)
self.assertTrue(len(record["PartitionKey"]) <= 256)

def test_create_records(self):
test_data_count = 10
Expand All @@ -23,11 +23,11 @@ def test_create_records(self):
records = kinesis.create_records(data_list)

for i, record in enumerate(records):
self.assertTrue('Data' in record)
self.assertTrue('PartitionKey' in record)
self.assertTrue("Data" in record)
self.assertTrue("PartitionKey" in record)

# ensure same order
self.assertEqual(record['Data'].decode(), data_list[i])
self.assertEqual(record["Data"].decode(), data_list[i])

self.assertTrue(type(record['PartitionKey']) == str)
self.assertTrue(len(record['PartitionKey']) <= 256)
self.assertTrue(type(record["PartitionKey"]) == str)
self.assertTrue(len(record["PartitionKey"]) <= 256)

0 comments on commit e24dff1

Please sign in to comment.