Skip to content

Commit

Permalink
yapf, typo in test
Browse files Browse the repository at this point in the history
  • Loading branch information
TheNeuralBit committed Aug 30, 2022
1 parent 54c5a62 commit c088431
Showing 1 changed file with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -169,8 +169,10 @@ def _split_batch_into_n_partitions(self, N):
]

lengths = [len(element_batch) for element_batch in element_batches]
batches = [self.converter.produce_batch(element_batch)
for element_batch in element_batches]
batches = [
self.converter.produce_batch(element_batch)
for element_batch in element_batches
]

return batches, lengths

Expand All @@ -192,13 +194,12 @@ def test_combine_batches(self, N):
(3, ),
(10, ),
])
def test_get_lenth(self, N):
def test_get_length(self, N):
batches, lengths = self._split_batch_into_n_partitions(N)

for batch, expected_length in zip(batches, lengths):
self.assertEqual(self.converter.get_length(batch), expected_length)


def test_equals(self):
self.assertTrue(self.converter == self.create_batch_converter())
self.assertTrue(self.create_batch_converter() == self.converter)
Expand Down

0 comments on commit c088431

Please sign in to comment.