def test_div(self): for key in self.arrays: arr, nzarr1 = self.get_arrs(key) for key2 in self.arrays: arr2, nzarr2 = self.get_arrs(key2) sumnp = div0(arr, arr2) sumnz = nzarr1 / nzarr2 summix = nzarr1 / arr2 self.assertTrue((sumnz.full() == sumnp).all(), f"{key} failed") self.assertTrue((summix.full() == sumnp).all(), f"{key} failed") self.assertTrue( ((nzarr1 / 2).full() == arr / 2).all(), f"{key} failed" )
def test_floordiv(self): for key in self.arrays: nzarr1, arr = self.get_arrs(key) for key2 in self.arrays: nzarr2, arr2 = self.get_arrs(key2) sumnp = div0(arr, arr2).astype(int) sumnz = nzarr1 // nzarr2 summix = nzarr1 // arr2 self.assertTrue((sumnz.full() == sumnp).all(), f"{key} failed") self.assertTrue((summix.full() == sumnp).all(), f"{key} failed") self.assertTrue( ((nzarr1 // 2).full() == arr // 2).all(), f"{key} failed" )
def test_floordiv(self): for key, arr in self.arrays.items(): for key2, arr2 in self.arrays.items(): nzarr1 = nzarray2d(arr) nzarr2 = nzarray2d(arr2) sumnp = div0(arr, arr2).astype(int) sumnz = nzarr1 // nzarr2 summix = nzarr1 // arr2 self.assertTrue((sumnz.full() == sumnp).all(), f"{key} failed") self.assertTrue((summix.full() == sumnp).all(), f"{key} failed") self.assertTrue( ((nzarr1 // 2).full() == arr // 2).all(), f"{key} failed" )