lass_j/AMBER
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 4 Wiki Activity

Make flake8 linting

Browse Source 
Add gitignore
This commit is contained in:
Jakob Lass
2025-04-09 17:27:28 +02:00
parent c689f35b76
commit e57a01b9a1
4 changed files with 83 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_background.py
View File

@@ -1,8 +1,6 @@
import unittest
import torch
import numpy as np
import os
import sys
maindir = os.getcwd()
@@ -11,7 +9,6 @@ sys.path.append(main_path+"/ds4ms/code/src")
from background import background
class TestBackground(unittest.TestCase):
def setUp(self):
@@ -19,7 +16,7 @@ class TestBackground(unittest.TestCase):
self.bg.load_data(verbose=True)
self.bg.set_grid_volume(dqx=0.03, dqy=0.03, dE=0.08)
self.bg.set_binned_data()
self.bg.set_radial_bins(max_radius=6.0, n_bins=10
self.bg.set_radial_bins(max_radius=6.0, n_bins=10)
self.bg.Ygrid = torch.tensor(np.random.rand(10, 10), dtype=torch.float64)
def test_load_data(self):
@@ -166,12 +163,14 @@ class TestBackground(unittest.TestCase):
alpha_range = torch.tensor([1.0])
beta_range = torch.tensor([1.0])
mu_range = torch.tensor([1.0])
result = self.bg.cross_validation(lambda_range=lambda_range, alpha_range=alpha_range, beta_range=beta_range, mu_range=mu_range, n_epochs=1, verbose=False)
result = self.bg.cross_validation(lambda_range=lambda_range, alpha_range=alpha_range,
beta_range=beta_range, mu_range=mu_range, n_epochs=1, verbose=False)
self.assertIsNotNone(result)
def test_compute_mask(self):
result = self.bg.compute_mask(q=0.75, e_cut=None)
self.assertIsNotNone(result)
if __name__ == '__main__':
unittest.main()
unittest.main()

10
tests/test_graph_laplacian.py
View File

@@ -2,14 +2,15 @@ import unittest
import numpy as np
import torch
from scipy.sparse import lil_matrix, csr_matrix, coo_matrix
import os
import sys
maindir = os.getcwd()
main_path = maindir[:maindir.find('ds4ms/code')]
sys.path.append(main_path+"/ds4ms/code/src")
from graph_laplacian import create_laplacian_matrix, delete_from_csr, remove_vertex, laplacian, unnormalized_laplacian, laplacian_chain, unnormalized_laplacian_chain
from graph_laplacian import create_laplacian_matrix, delete_from_csr, \
remove_vertex, laplacian, unnormalized_laplacian, laplacian_chain, \
unnormalized_laplacian_chain
class TestGraphLaplacian(unittest.TestCase):
@@ -66,5 +67,6 @@ class TestGraphLaplacian(unittest.TestCase):
self.assertEqual(L.shape, (nx, nx))
self.assertIsInstance(L, csr_matrix)
if __name__ == '__main__':
unittest.main()
unittest.main()