tests-and-examples-folders

open2c · Oct 3, 2023 · 0ea23c8 · 0ea23c8
1 parent 7e4a410
commit 0ea23c8
Show file tree

Hide file tree

Showing 5 changed files with 616 additions and 605 deletions.
diff --git a/examples/sample_simulation_onesided.ipynb b/examples/sample_simulation_onesided.ipynb
diff --git a/looplib/sample_simulation_onesided.ipynb b/looplib/sample_simulation_onesided.ipynb
diff --git a/looplib/test.py b/looplib/test.py
diff --git a/setup.py b/setup.py
@@ -14,6 +14,7 @@
     install_requires=['numpy', 'matplotlib'],
     ext_modules = cythonize(['looplib/simlef.pyx',
                              'looplib/simlef_twosided.pyx',
+                             'looplib/simlef_onesided.pyx',
                              'looplib/looptools_c.pyx']),
     include_dirs=[np.get_include()]
 )
diff --git a/tests/test_looptools.py b/tests/test_looptools.py
@@ -0,0 +1,16 @@
+from looplib.looptools import *
+
+
+def test_FRiP():
+    # Tests for FRiP
+    lattice_length = 5
+    boundary_list = [1, 4]
+
+    lef_A = [1, 1, 4, 1]
+    assert FRiP(lattice_length, lef_A, boundary_list) == 1
+
+    lef_B = [3, 0, 2, 3]
+    assert FRiP(lattice_length, lef_B, boundary_list) == 0
+
+    lef_C = [1, 1, 3, 3]
+    assert FRiP(lattice_length, lef_C, boundary_list) == 0.5