tests-and-examples-folders

Author: gfudenberg

examples/sample_simulation_onesided.ipynb

@@ -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()]
 )

looplib/sample_simulation_onesided.ipynb

@@ -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