modified GitHub workflow action script test.yml

Author: ParticularMiner

.github/workflows/test.yml

@@ -22,7 +22,11 @@ jobs:
         python-version: ${{ matrix.python-version }}
 
     - name: Install package
-      run: pip install .
+      run: |
+        python -m pip install --upgrade pip
+        pip install -e .
     
     - name: Run tests
-      run: python -m unittest
+      run: |
+        pip install pytest
+        pytest -ra --capture=no --showlocals

setup.py

@@ -31,8 +31,7 @@ def finalize_options(self):
 
 array_wrappers_ext = Extension('sparse_dot_topn.array_wrappers',
                          sources=[
-                                    './sparse_dot_topn/array_wrappers.pyx',
-                                    './sparse_dot_topn/sparse_dot_topn_source.cpp'
+                                    './sparse_dot_topn/array_wrappers.pyx'
                                 ],
                          extra_compile_args=extra_compile_args,
                          language='c++')

sparse_dot_topn/sparse_dot_topn_parallel.cpp

@@ -20,6 +20,7 @@
 // Modified by: Particular Miner
 // April 14, 2021
 
+#include <cmath>
 #include <vector>
 #include <algorithm>
 #include <numeric>
@@ -398,7 +399,7 @@ int sparse_dot_topn_extd_parallel(
 	std::vector<int> sub_total(n_jobs, 0);
 	std::vector<int> split_n_minmax(n_jobs, 0);
 
-	int mem_sz_per_row = std::max(1, (int) ceil(((double) nnz_max)/((double) n_row)));
+	int mem_sz_per_row = std::max(1, (int) std::ceil(((double) nnz_max)/((double) n_row)));
 
 	std::vector<std::thread> thread_list(n_jobs);
 

sparse_dot_topn/test/test_awesome_cossim_topn.py

@@ -280,7 +280,7 @@ def test_awesome_cossim_top_large_matrix():
     # test with large matrix
     nr_vocab = 2 << 24
     density = 1e-6
-    n_samples = 10000
+    n_samples = 1000
     nnz = int(n_samples * nr_vocab * density)
 
     rng1 = np.random.RandomState(42)