From f2898fd369c6efea2ea7f0fd6da2a2a8033ea3d6 Mon Sep 17 00:00:00 2001 From: Hai Nguyen Date: Fri, 22 Nov 2024 14:08:00 -0500 Subject: [PATCH] more --- nglview/tests/test_widget.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/nglview/tests/test_widget.py b/nglview/tests/test_widget.py index 9f0db547..f1ef2b91 100644 --- a/nglview/tests/test_widget.py +++ b/nglview/tests/test_widget.py @@ -95,10 +95,17 @@ def __init__(self): @property def n_frames(self): - return 10 + return 5 def get_coordinates(self, frame): - return np.random.rand(10, 3, 3) + coordinates = [ + [[0.0, 0.0, 0.0], [1.0, 0.0, 0.0], [0.0, 1.0, 0.0]], + [[0.1, 0.1, 0.1], [1.1, 0.1, 0.1], [0.1, 1.1, 0.1]], + [[0.2, 0.2, 0.2], [1.2, 0.2, 0.2], [0.2, 1.2, 0.2]], + [[0.3, 0.3, 0.3], [1.3, 0.3, 0.3], [0.3, 1.3, 0.3]], + [[0.4, 0.4, 0.4], [1.4, 0.4, 0.4], [0.4, 1.4, 0.4]] + ] + return np.array(coordinates)[frame] def get_structure_string(self): return 'hello'