lass_j/AMBER
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 1 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()