python3: monkeypatching variations in import-export site cmd testing

tests/test_import_export_site_cmd.py

@@ -17,6 +17,7 @@
 import random
 import json
 
+from django.utils import six
 from django.utils.six.moves import builtins as __builtin__
 from django.core import management
 import pytest
@@ -165,7 +166,8 @@ def test_import_site_confirm_prompt_yes(db, monkeypatch, json_fixture):
     def yes_raw_input(*args, **kwargs):
         return 'yes'
 
-    monkeypatch.setattr(__builtin__, 'raw_input', yes_raw_input)
+    input_funcname = 'raw_input' if six.PY2 else 'input'
+    monkeypatch.setattr(__builtin__, input_funcname, yes_raw_input)
 
     management.call_command('import_site', json_fixture(content), stdin='yes')
     assert Role.objects.get(uuid='dqfewrvesvews2532')