Merge pull request #16 from asmeurer/fix-tests

Fix __all__ for numpy and cupy

Author: asmeurer

array_api_compat/common/_aliases.py

@@ -408,7 +408,8 @@ def trunc(x: ndarray, /, xp, **kwargs) -> ndarray:
         return x
     return xp.trunc(x, **kwargs)
 
-__all__ = ['UniqueAllResult', 'UniqueCountsResult', 'UniqueInverseResult',
+__all__ = ['arange', 'empty', 'empty_like', 'eye', 'full', 'full_like',
+           'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like', 'UniqueAllResult', 'UniqueCountsResult', 'UniqueInverseResult',
            'unique_all', 'unique_counts', 'unique_inverse', 'unique_values',
            'astype', 'std', 'var', 'permute_dims', 'reshape', 'argsort',
            'sort', 'sum', 'prod', 'ceil', 'floor', 'trunc']

array_api_compat/cupy/_aliases.py

@@ -58,6 +58,7 @@
 floor = get_xp(cp)(_aliases.floor)
 trunc = get_xp(cp)(_aliases.trunc)
 
-__all__ = _aliases.__all__ + ['asarray', 'asarray_cupy', 'bool', 'arange',
-                              'empty', 'empty_like', 'eye', 'full', 'full_like',
-                              'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like']
+__all__ = _aliases.__all__ + ['asarray', 'asarray_cupy', 'bool', 'acos',
+                              'acosh', 'asin', 'asinh', 'atan', 'atan2',
+                              'atanh', 'bitwise_left_shift', 'bitwise_invert',
+                              'bitwise_right_shift', 'concat', 'pow']

array_api_compat/numpy/_aliases.py

@@ -58,6 +58,7 @@
 floor = get_xp(np)(_aliases.floor)
 trunc = get_xp(np)(_aliases.trunc)
 
-__all__ = _aliases.__all__ + ['asarray', 'asarray_numpy', 'bool', 'arange',
-                              'empty', 'empty_like', 'eye', 'full', 'full_like',
-                              'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like']
+__all__ = _aliases.__all__ + ['asarray', 'asarray_numpy', 'bool', 'acos',
+                              'acosh', 'asin', 'asinh', 'atan', 'atan2',
+                              'atanh', 'bitwise_left_shift', 'bitwise_invert',
+                              'bitwise_right_shift', 'concat', 'pow']