• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• RBF-master
  • environment.yml
  • readthedocs.yml
  • conda.recipe
    • run_test.py
    • conda_build_config.yaml
    • meta.yaml
    • build.sh
  • test
    • test_utils.py
    • test_linalg.py
    • test_gauss.py
    • test_fd.py
    • test_geometry.py
    • test_interpolate.py
    • test_sputils.py
    • test_basis.py
    • test_domain.py
  • README.rst
  • setup.py
  • rbf
    • gauss.py
    • poly.pyx
    • pde
      • fd.py
      • elastic.py
      • nodes.py
      • halton.pyx
      • domain.py
      • sampling.pyx
      • geometry.pyx
      • __init__.py
    • basis.py
    • misc
      • mp.py
      • octtree.pyx
      • README.rst
      • quadtree.pyx
      • filter.py
      • integrate.py
      • __init__.py
      • bspline.pyx
      • formulation.py
    • linalg.py
    • sputils.pyx
    • __init__.py
    • utils.py
    • interpolate.py
  • make_version.py
  • requirements.txt
  • .gitignore
  • docs
    • Makefile
    • source
      • pde.rst
      • linalg.rst
      • poly.rst
      • basis.rst
      • interpolate.rst
      • geometry.rst
      • fd.rst
      • utils.rst
      • halton.rst
      • sampling.rst
      • gauss.rst
      • nodes.rst
      • sputils.rst
      • index.rst
      • conf.py
      • installation.rst
    • figures
    • scripts
      • gauss.e.py
      • fd.l.py
      • fd.i.py
      • fd.b.py
      • gauss.i.py
      • nodes.b.py
      • gauss.c.py
      • basis.a.py
      • gauss.g.py
      • fd.c.py
      • sampling.b.py
      • gauss.b.py
      • gauss.k.py
      • gauss.d.py
      • nodes.a.py
      • fd.j.py
      • sampling.a.py
      • basis.b.py
      • interpolate.b.py
      • fd.k.py
      • gauss.f.py
      • fd.d.py
      • gauss.j.py
      • interpolate.a.py
      • gauss.a.py
  • LICENSE.txt
  • TODO.txt

import numpy as np
import scipy.sparse as sp
import scipy.sparse.linalg as spla
import rbf.sputils
import unittest

class Test(unittest.TestCase):
  def test_row_norms(self):
    A = sp.rand(100, 100, 0.1)
    A = A.tocsc()

    n1 = spla.norm(A, axis=1, ord=2)
    n2 = rbf.sputils.row_norms(A, order=2)
    self.assertTrue(np.allclose(n1, n2))

    n1 = spla.norm(A, axis=1, ord=1)
    n2 = rbf.sputils.row_norms(A, order=1)
    self.assertTrue(np.allclose(n1, n2))

  def test_divide_rows(self):
    A = sp.rand(100, 10, 0.1)
    
    n = np.random.uniform(1.0, 2.0, (100,))
    B1 = rbf.sputils.divide_rows(A, n)
    B2 = sp.diags(1.0/n).dot(A)
    
    self.assertTrue(np.allclose(B1.A, B2.A))

  def test_divide_rows_inplace(self):
    A = sp.rand(100, 10, 0.1)
    A = A.tocsc()

    n = np.random.uniform(1.0, 2.0, (100,))
    B = sp.diags(1.0/n).dot(A)
    # This should modify A inplace as long as A is csc, csr, or coo
    rbf.sputils.divide_rows(A, n, inplace=True)
    self.assertTrue(np.allclose(A.A, B.A))

  def test_add_rows(self):
    A = sp.rand(100, 10, 1.0)
    B = sp.rand(20, 10, 1.0)
    idx = np.random.choice(range(100), size=(20,), replace=False)
    C = rbf.sputils.add_rows(A, B, idx)

    # this should be equivalent to `add_rows`
    D = A.tocsc(copy=True)
    D[idx, :] += B
    
    self.assertTrue(np.allclose(C.A, D.A))