Merge pull request qutip#2038 from hodgestar/fix/add-back-qu-suffix-t…

…o-saved-objects

Add back .qu suffix to saved objects

doc/changes/2038.bugfix

@@ -0,0 +1 @@
+Add back .qu suffix to objects saved with qsave and loaded with qload.

qutip/fileio.py

@@ -221,10 +221,10 @@ def qsave(data, name='qutip_data'):
 
     """
     # open the file for writing
-    file = Path(name)
-    file = file.with_suffix(file.suffix + ".qu")
+    path = Path(name)
+    path = path.with_suffix(path.suffix + ".qu")
 
-    with open(name, "wb") as fileObject:
+    with open(path, "wb") as fileObject:
         # this writes the object a to the file named 'filename.qu'
         pickle.dump(data, fileObject)
 
@@ -244,10 +244,10 @@ def qload(name):
         Object retrieved from requested file.
 
     """
-    file = Path(name)
-    file = file.with_suffix(file.suffix + ".qu")
+    path = Path(name)
+    path = path.with_suffix(path.suffix + ".qu")
 
-    with open(name, "rb") as fileObject:
+    with open(path, "rb") as fileObject:
         if sys.version_info >= (3, 0):
             out = pickle.load(fileObject, encoding='latin1')
         else:

qutip/tests/test_fileio.py

@@ -13,6 +13,7 @@
 
 _dimension = 10
 
+
 def _random_file_name():
     return "_" + str(uuid.uuid4())
 
@@ -57,3 +58,5 @@ def test_qsave_qload(use_path, suffix):
     qutip.qsave(ops_in, filename)
     ops_out = qutip.qload(filename)
     assert ops_in == ops_out
+    # check that the file was saved with the correct name:
+    assert Path(str(filename) + ".qu").exists()