def test__randomize_samples_false(): """Test that ``_randomize_samples`` is a no-op when user wants random samples. Input: - randomize_samples as False """ # Setup instance = Mock() randomize_samples = False # Run BaseTabularModel._randomize_samples(instance, randomize_samples) # Assert assert instance._set_random_state.called_once_with(None)
def test__randomize_samples_true(): """Test that ``_randomize_samples`` sets the random state correctly. Input: - randomize_samples as True Side Effect: - random state is set """ # Setup instance = Mock() randomize_samples = True # Run BaseTabularModel._randomize_samples(instance, randomize_samples) # Assert assert instance._set_random_state.called_once_with(FIXED_RNG_SEED)