Repository: zapata-engineering/orqviz (language: python)

Issue:
namespace for tests imports



Discussion / hints:

Fix patch:
diff --git a/README.md b/README.md
index 25f4913..792d30a 100644
--- a/README.md
+++ b/README.md
@@ -6,6 +6,7 @@ A Python package for easily **visualizing the loss landscape** of Variational Qu
 ## Getting started
 
 In our repository you will find an [example Jupter notebook](https://github.com/zapatacomputing/orqviz/blob/main/examples/example_notebook.ipynb) providing a thorough demonstration of the tools available with the package. 
+
 In [this Github](https://github.com/zapatacomputing/visualization-tutorials) repository we have four Jupyter notebooks with tutorials for how to get started with any quantum circuit simulation framework you might use. You will find examples with `qiskit`, `cirq`, `pennylane` and Zapata's `Orquestra` library. The tutorials are not exhaustive, but they do provide a full story that you can follow along.
 
 We have recently published a paper on arXiv where we review the tools available with `orqviz`. TODO: Link to arXiv.
@@ -31,7 +32,7 @@ import numpy as np
 np.random.seed(42)
 
 def loss_function(pars):
-    return np.sum(np.cos(pars))**2
+    return np.sum(np.cos(pars))**2 + np.sum(np.sin(30*pars))**2
 
 n_params = 42
 params = np.random.uniform(-np.pi, np.pi, size=n_params)
@@ -40,12 +41,25 @@ dir2 = orqviz.geometric.get_random_orthonormal_vector(dir1)
 
 scan2D_result = orqviz.scans.perform_2D_scan(params, loss_function, 
                                 direction_x=dir1, direction_y=dir2,
-                                n_steps_x=60)
+                                n_steps_x=100)
 orqviz.scans.plot_2D_scan_result(scan2D_result)
 ```
-This code results in the following plot:\
+This code results in the following plot:
+
+
 ![Image](docs/example_plot.png)
 
+## Authors
+The leading developer of this package is Manuel Rudolph at Zapata Computing.\
+For questions related to the visualization techniques, contact Manuel via Manuel.Rudolph@zapatacomputing.com
+
+The leading software developer of this package is Michal Stechly at Zapata Computing.\
+For questions related to technicalities of the package, contact Michal via Michal.Stechly@zapatacomputing.com
+
+For general questions, problems, bug fixes, etc. please open a [GitHub issue](https://github.com/zapatacomputing/orqviz/issues) in the `orqviz` repository.
+
+Thank you to Sukin Sim and Luis Serrano from Zapata Computing for their contribution to the tutorials.
+
 ## Further
 TODO:
 Troubleshooting\
diff --git a/docs/example_plot.png b/docs/example_plot.png
index 74a9110..37d15f9 100644
Binary files a/docs/example_plot.png and b/docs/example_plot.png differ


Test patch:
diff --git a/tests/orqviz/elastic_band_test.py b/tests/orqviz/elastic_band_test.py
index 51729fb..46f6ca0 100644
--- a/tests/orqviz/elastic_band_test.py
+++ b/tests/orqviz/elastic_band_test.py
@@ -1,13 +1,13 @@
 import pytest
 import numpy as np
 import matplotlib.pyplot as plt
-from zquantum.visualization.elastic_band import (
+from orqviz.elastic_band import (
     Chain,
     run_NEB,
     run_AutoNEB,
     plot_all_chains_losses,
 )
-from zquantum.visualization.utils import save_viz_object, load_viz_object
+from orqviz.utils import save_viz_object, load_viz_object
 
 
 def SUM_OF_SINS(params):
diff --git a/tests/orqviz/hessians_test.py b/tests/orqviz/hessians_test.py
index c21f2a8..3989ab8 100644
--- a/tests/orqviz/hessians_test.py
+++ b/tests/orqviz/hessians_test.py
@@ -1,14 +1,14 @@
 import pytest
 import numpy as np
 import matplotlib.pyplot as plt
-from zquantum.visualization.hessians import (
+from orqviz.hessians import (
     get_Hessian,
     get_Hessian_SPSA_approx,
     HessianEigenobject,
     perform_1D_hessian_eigenvector_scan,
     plot_1D_hessian_eigenvector_scan_result,
 )
-from zquantum.visualization.utils import save_viz_object, load_viz_object
+from orqviz.utils import save_viz_object, load_viz_object
 
 
 def COST_FUNCTION(params):
diff --git a/tests/orqviz/pca_test.py b/tests/orqviz/pca_test.py
index b39469e..0ffa044 100644
--- a/tests/orqviz/pca_test.py
+++ b/tests/orqviz/pca_test.py
@@ -1,7 +1,7 @@
 import pytest
 import numpy as np
-from zquantum.visualization.scans.data_structures import Scan2DResult
-from zquantum.visualization.pca import (
+from orqviz.scans.data_structures import Scan2DResult
+from orqviz.pca import (
     perform_2D_pca_scan,
     get_pca,
     PCAobject,
diff --git a/tests/orqviz/scans_test.py b/tests/orqviz/scans_test.py
index 0095e2f..0142786 100644
--- a/tests/orqviz/scans_test.py
+++ b/tests/orqviz/scans_test.py
@@ -1,21 +1,21 @@
 import pytest
 import numpy as np
-from zquantum.visualization.scans import (
+from orqviz.scans import (
     perform_1D_scan,
     perform_1D_interpolation,
     perform_2D_scan,
     perform_2D_interpolation,
     get_2D_slice_around_point,
 )
-from zquantum.visualization.scans.plots import (
+from orqviz.scans.plots import (
     plot_1D_scan_result,
     plot_2D_scan_result,
     plot_2D_interpolation_result,
 )
-from zquantum.visualization.scans.data_structures import Scan1DResult, Scan2DResult
+from orqviz.scans.data_structures import Scan1DResult, Scan2DResult
 import matplotlib.pyplot as plt
-from zquantum.visualization.scans import perform_1D_scan
-from zquantum.visualization.utils import save_viz_object, load_viz_object
+from orqviz.scans import perform_1D_scan
+from orqviz.utils import save_viz_object, load_viz_object
 
 
 def SUM_OF_SINS(params):
@@ -23,8 +23,8 @@ def SUM_OF_SINS(params):
 
 
 def test_1D_scan():
-    origin = np.array([0, 0])
-    direction_x = np.array([1, 0])
+    origin = np.array([0.0, 0.0])
+    direction_x = np.array([1.0, 0.0])
     end_points_x = (-np.pi, np.pi)
     n_steps_x = 100
 
@@ -59,8 +59,8 @@ def test_1D_scan():
 
 
 def test_1D_interpolation():
-    point1 = np.array([0, 0])
-    point2 = np.array([0, 0])
+    point1 = np.array([1, 0])
+    point2 = np.array([0, 1])
     direction_x = point2 - point1
     end_points_x = (0, 1)
     n_steps_x = 23
@@ -78,7 +78,7 @@ def test_1D_interpolation():
     for scan_results in [scan_1d, loaded_scan1d]:
         assert isinstance(scan_results, Scan1DResult)
         assert len(scan_results.values) == n_steps_x
-        np.testing.assert_equal(np.dot(scan_results.direction, direction_x), 0)
+        np.testing.assert_equal(scan_results.direction, direction_x)
         assert scan_results.params_list.shape == (n_steps_x, 2)
 
         np.testing.assert_equal(
@@ -177,7 +177,7 @@ def test_get_2D_slice_around_point_in_5D_space():
 
 
 def test_interpolation_2D_in_2D_space():
-    point_1 = np.array([0, 0])
+    point_1 = np.array([1, 0])
     point_2 = np.array([0, 2])
     end_points_x = (-0.5, 1.5)
     end_points_y = (-0.5, 0.5)
diff --git a/tests/orqviz/utils_test.py b/tests/orqviz/utils_test.py
index ddb70a4..98168fe 100644
--- a/tests/orqviz/utils_test.py
+++ b/tests/orqviz/utils_test.py
@@ -1,11 +1,11 @@
 import pytest
 import numpy as np
-from zquantum.visualization.scans import perform_2D_scan, perform_1D_scan
-from zquantum.visualization.hessians import get_Hessian
-from zquantum.visualization.utils import save_viz_object, load_viz_object, OrqVizObject
-from zquantum.visualization.scans.data_structures import Scan1DResult, Scan2DResult
-from zquantum.visualization.hessians.data_structures import HessianEigenobject
-from zquantum.visualization.elastic_band.data_structures import Chain
+from orqviz.scans import perform_2D_scan, perform_1D_scan
+from orqviz.hessians import get_Hessian
+from orqviz.utils import save_viz_object, load_viz_object, OrqVizObject
+from orqviz.scans.data_structures import Scan1DResult, Scan2DResult
+from orqviz.hessians.data_structures import HessianEigenobject
+from orqviz.elastic_band.data_structures import Chain
 
 
 def SUM_OF_SINS(params):