def test_from_params(self): params = Params({}) sorting_keys = [("s1", "nt"), ("s2", "nt2")] params["sorting_keys"] = sorting_keys iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.1 assert not iterator._biggest_batch_first assert iterator._batch_size == 32 assert not iterator._skip_smaller_batches params = Params( { "sorting_keys": sorting_keys, "padding_noise": 0.5, "biggest_batch_first": True, "batch_size": 100, "skip_smaller_batches": True, } ) iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.5 assert iterator._biggest_batch_first assert iterator._batch_size == 100 assert iterator._skip_smaller_batches
def test_from_params(self): # pylint: disable=protected-access params = Params({}) with pytest.raises(ConfigurationError): iterator = BucketIterator.from_params(params) sorting_keys = [("s1", "nt"), ("s2", "nt2")] params['sorting_keys'] = sorting_keys iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.1 assert not iterator._biggest_batch_first assert iterator._batch_size == 32 params = Params({ "sorting_keys": sorting_keys, "padding_noise": 0.5, "biggest_batch_first": True, "batch_size": 100 }) iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.5 assert iterator._biggest_batch_first assert iterator._batch_size == 100
def test_from_params(self): # pylint: disable=protected-access params = Params({}) with pytest.raises(ConfigurationError): iterator = BucketIterator.from_params(params) sorting_keys = [("s1", "nt"), ("s2", "nt2")] params['sorting_keys'] = sorting_keys iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.1 assert not iterator._biggest_batch_first assert iterator._batch_size == 32 params = Params({ "sorting_keys": sorting_keys, "padding_noise": 0.5, "biggest_batch_first": True, "batch_size": 100 }) iterator = BucketIterator.from_params(params) assert iterator._sorting_keys == sorting_keys assert iterator._padding_noise == 0.5 assert iterator._biggest_batch_first assert iterator._batch_size == 100