Use more robust test in test_inputs.py too