def test__reset_primary_keys_generators(self):
"""Test reset values"""
# Run
sampler = Mock()
sampler.primary_key = 'something'
sampler.remaining_primary_key = 'else'
Sampler._reset_primary_keys_generators(sampler)
# Asserts
assert sampler.primary_key == dict()
assert sampler.remaining_primary_key == dict()
def test__reset_primary_keys_generators(self):
"""_reset_primary_keys deletes all generators and counters."""
# Setup
data_navigator = MagicMock(spec=DataNavigator)
modeler = MagicMock(spec=Modeler)
sampler = Sampler(data_navigator=data_navigator, modeler=modeler)
sampler.primary_key = {'table': 'generator for table'}
sampler.remaining_primary_key = {'table': 'counter for table'}
# Run
sampler._reset_primary_keys_generators()
# Check
assert sampler.primary_key == dict()
assert sampler.remaining_primary_key == dict()
