def test_write_file(with_open):
path = Mock()
data = Mock()
write_file(path, data)
with_open.assert_called_once_with(path, "w")
with_open().write.assert_called_once_with(data)
def test_write_file_mode(with_open):
path = Mock()
data = Mock()
mode = Mock()
write_file(path, data, mode)
with_open.assert_called_once_with(path, mode)
with_open().write.assert_called_once_with(data)
def _create_model_server_config_file():
configuration_properties = _generate_mms_config_properties()
utils.write_file(MMS_CONFIG_FILE, configuration_properties)
def _create_torchserve_config_file():
configuration_properties = _generate_ts_config_properties()
utils.write_file(TS_CONFIG_FILE, configuration_properties)
