added more viewer tests and use from larray import a, b, c to shorten them

Author: gdementen

larray_editor/api.py

@@ -268,20 +268,20 @@ def restore_display_hook():
 if __name__ == "__main__":
     """Array editor test"""
     import numpy as np
+    from larray import (Session, Axis, LArray, ndtest, ndrange, zeros, from_lists, union,
+                        sin, cos, radians, maximum, sqrt)
 
-    lipro = la.Axis(['P%02d' % i for i in range(1, 16)], 'lipro')
-    age = la.Axis('age=0..115')
-    sex = la.Axis('sex=M,F')
+    lipro = Axis(['P%02d' % i for i in range(1, 16)], 'lipro')
+    age = Axis('age=0..115')
+    sex = Axis('sex=M,F')
 
-    vla = 'A11,A12,A13,A23,A24,A31,A32,A33,A34,A35,A36,A37,A38,A41,A42,' \
-          'A43,A44,A45,A46,A71,A72,A73'
-    wal = 'A25,A51,A52,A53,A54,A55,A56,A57,A61,A62,A63,A64,A65,A81,A82,' \
-          'A83,A84,A85,A91,A92,A93'
+    vla = 'A11,A12,A13,A23,A24,A31,A32,A33,A34,A35,A36,A37,A38,A41,A42,A43,A44,A45,A46,A71,A72,A73'
+    wal = 'A25,A51,A52,A53,A54,A55,A56,A57,A61,A62,A63,A64,A65,A81,A82,A83,A84,A85,A91,A92,A93'
     bru = 'A21'
     # list of strings
-    belgium = la.union(vla, wal, bru)
+    belgium = union(vla, wal, bru)
 
-    geo = la.Axis(belgium, 'geo')
+    geo = Axis(belgium, 'geo')
 
     # data1 = np.arange(30).reshape(2, 15)
     # arr1 = la.LArray(data1, axes=(sex, lipro))
@@ -295,8 +295,8 @@ def restore_display_hook():
     # data2 = np.random.randint(10, size=(116, 44, 2, 15)) / 100 + 1567
     # data2 = np.random.normal(51000000, 10000000, size=(116, 44, 2, 15))
     data2 = np.random.normal(0, 1, size=(116, 44, 2, 15))
-    arr2 = la.LArray(data2, axes=(age, geo, sex, lipro))
-    # arr2 = la.ndrange([100, 100, 100, 100, 5])
+    arr2 = LArray(data2, axes=(age, geo, sex, lipro))
+    # arr2 = ndrange([100, 100, 100, 100, 5])
     # arr2 = arr2['F', 'A11', 1]
 
     # view(arr2[0, 'A11', 'F', 'P01'])
@@ -314,59 +314,87 @@ def restore_display_hook():
     # print(arr2[0, 'A11', :, 'P01'])
 
     # data2 = np.random.normal(0, 10.0, size=(5000, 20))
-    # arr2 = la.LArray(data2,
-    #                  axes=(la.Axis(list(range(5000)), 'd0'),
-    #                        la.Axis(list(range(20)), 'd1')))
+    # arr2 = LArray(data2, axes=(Axis(list(range(5000)), 'd0'),
+    #                            Axis(list(range(20)), 'd1')))
     # edit(arr2)
 
     # view(['a', 'bb', 5599])
     # view(np.arange(12).reshape(2, 3, 2))
     # view([])
 
     data3 = np.random.normal(0, 1, size=(2, 15))
-    arr3 = la.ndrange((30, sex))
+    arr3 = ndrange((30, sex))
     # data4 = np.random.normal(0, 1, size=(2, 15))
-    # arr4 = la.LArray(data4, axes=(sex, lipro))
+    # arr4 = LArray(data4, axes=(sex, lipro))
 
     # arr4 = arr3.copy()
     # arr4['F'] /= 2
-    arr4 = arr3.min(la.x.sex)
-    arr5 = arr3.max(la.x.sex)
-    arr6 = arr3.mean(la.x.sex)
+    arr4 = arr3.min(sex)
+    arr5 = arr3.max(sex)
+    arr6 = arr3.mean(sex)
 
     # test isssue #35
-    arr7 = la.from_lists([['a',                   1,                    2,                    3],
-                          [ '', 1664780726569649730, -9196963249083393206, -7664327348053294350]])
+    arr7 = from_lists([['a',                   1,                    2,                    3],
+                       [ '', 1664780726569649730, -9196963249083393206, -7664327348053294350]])
+
+    def make_circle(width=20, radius=9):
+        x, y = Axis(width, 'x'), Axis(width, 'y')
+        center = (width - 1) / 2
+        return maximum(radius - sqrt((x - center) ** 2 + (y - center) ** 2), 0)
+
+    def make_sphere(width=20, radius=9):
+        x, y, z = Axis(width, 'x'), Axis(width, 'y'), Axis(width, 'z')
+        center = (width - 1) / 2
+        return maximum(radius - sqrt((x - center) ** 2 + (y - center) ** 2 + (z - center) ** 2), 0)
+
+    def make_demo(width=20, ball_radius=5, path_radius=5, steps=30):
+        x, y = Axis(width, 'x'), Axis(width, 'y')
+        t = Axis(steps, 't')
+        center = (width - 1) / 2
+        ball_center_x = sin(radians(t * 360 / steps)) * path_radius + center
+        ball_center_y = cos(radians(t * 360 / steps)) * path_radius + center
+        return maximum(ball_radius - sqrt((x - ball_center_x) ** 2 + (y - ball_center_y) ** 2), 0).transpose(x, y)
+
+    demo = make_demo(9, 2.5, 1.5)
+    sphere = make_sphere(9, 4)
+    extreme_array = LArray([-np.inf, -1, 0, np.nan, 1, np.inf])
+    scalar = LArray(0)
+    arr_empty = LArray([])
+    arr_obj = ndtest((2, 3)).astype(object)
+    arr_str = ndtest((2, 3)).astype(str)
+    big = ndtest((1000, 1000, 500))
 
     # test autoresizing
-    long_labels = la.zeros('a=a_long_label,another_long_label; b=this_is_a_label,this_is_another_one')
-    long_axes_names = la.zeros('first_axis=a0,a1; second_axis=b0,b1')
+    long_labels = zeros('a=a_long_label,another_long_label; b=this_is_a_label,this_is_another_one')
+    long_axes_names = zeros('first_axis=a0,a1; second_axis=b0,b1')
 
     # compare(arr3, arr4, arr5, arr6)
 
-    # view(la.stack((arr3, arr4), la.Axis('arrays=arr3,arr4')))
-    ses = la.Session(arr2=arr2, arr3=arr3, arr4=arr4, arr5=arr5, arr6=arr6, arr7=arr7, long_labels=long_labels,
-                     long_axes_names=long_axes_names, data2=data2, data3=data3)
+    # view(stack((arr3, arr4), Axis('arrays=arr3,arr4')))
+    # ses = Session(arr2=arr2, arr3=arr3, arr4=arr4, arr5=arr5, arr6=arr6, arr7=arr7, long_labels=long_labels,
+    #                  long_axes_names=long_axes_names, data2=data2, data3=data3)
 
     # from larray.tests.common import abspath
     # file = abspath('test_session.xlsx')
     # ses.save(file)
 
     # import cProfile as profile
-    # profile.runctx('edit(la.Session(arr2=arr2))', vars(), {},
+    # profile.runctx('edit(Session(arr2=arr2))', vars(), {},
     #                'c:\\tmp\\edit.profile')
-
-    # edit(ses)
+    edit()
     # edit(ses)
     # edit(file)
     # edit('fake_path')
     # edit(REOPEN_LAST_FILE)
 
-    edit(ses)
-    # compare(la.Session(arr2=arr2, arr3=arr3), la.Session(arr2=arr2 + 1.0, arr3=arr3 + 1.0))
-    # compare(arr3, arr3 + la.ndrange(arr3.axes))
+    # edit(arr2)
+    # compare(arr3, arr3 + ndrange(arr3.axes))
+    # compare(Session(arr4=arr4, arr3=arr3),
+    #         Session(arr4=arr4 + 1.0, arr3=arr3 * 2.0))
+    # compare(Session(arr2=arr2, arr3=arr3),
+    #         Session(arr2=arr2 + 1.0, arr3=arr3 * 2.0))
 
-    # s = la.local_arrays()
+    # s = local_arrays()
     # view(s)
     # print('HDF')
     # s.save('x.h5')
@@ -382,7 +410,3 @@ def restore_display_hook():
     # edit('x_csv')
 
     # compare(arr3, arr4, arr5, arr6)
-
-    # arr3 = la.ndrange((1000, 1000, 500))
-    # print(arr3.nbytes * 1e-9 + 'Gb')
-    # edit(arr3, minvalue=-99, maxvalue=25.123456)