Modify pina.__init__.py and rm useless .py files

* rm meta.py, plotter.py, writer.py
* modify __init__ file
* modify tests due to __init__ import
This commit is contained in:
Dario Coscia
2025-02-19 11:14:53 +01:00
committed by Nicola Demo
parent 9c9d4fe7e4
commit 810d215ca0
8 changed files with 12 additions and 485 deletions

View File

@@ -1,6 +1,6 @@
import torch
import pytest
from pina import TorchOptimizer
from pina.optim import TorchOptimizer
opt_list = [
torch.optim.Adam, torch.optim.AdamW, torch.optim.SGD, torch.optim.RMSprop

View File

@@ -1,75 +0,0 @@
from pina.domain import CartesianDomain
from pina import Condition, Plotter
from matplotlib.testing.decorators import image_comparison
import matplotlib.pyplot as plt
from pina.problem import SpatialProblem
from pina.equation import FixedValue
"""
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
TODO : Fix the tests once the Plotter class is updated
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
class FooProblem1D(SpatialProblem):
# assign output/ spatial and temporal variables
output_variables = ['u']
spatial_domain = CartesianDomain({'x' : [-1, 1]})
# problem condition statement
conditions = {
'D': Condition(location=CartesianDomain({'x': [-1, 1]}), equation=FixedValue(0.)),
}
class FooProblem2D(SpatialProblem):
# assign output/ spatial and temporal variables
output_variables = ['u']
spatial_domain = CartesianDomain({'x' : [-1, 1], 'y': [-1, 1]})
# problem condition statement
conditions = {
'D': Condition(location=CartesianDomain({'x' : [-1, 1], 'y': [-1, 1]}), equation=FixedValue(0.)),
}
class FooProblem3D(SpatialProblem):
# assign output/ spatial and temporal variables
output_variables = ['u']
spatial_domain = CartesianDomain({'x' : [-1, 1], 'y': [-1, 1], 'z':[-1,1]})
# problem condition statement
conditions = {
'D': Condition(location=CartesianDomain({'x' : [-1, 1], 'y': [-1, 1], 'z':[-1,1]}), equation=FixedValue(0.)),
}
def test_constructor():
Plotter()
def test_plot_samples_1d():
problem = FooProblem1D()
problem.discretise_domain(n=10, mode='grid', variables = 'x', locations=['D'])
pl = Plotter()
pl.plot_samples(problem=problem, filename='fig.png')
import os
os.remove('fig.png')
def test_plot_samples_2d():
problem = FooProblem2D()
problem.discretise_domain(n=10, mode='grid', variables = ['x', 'y'], locations=['D'])
pl = Plotter()
pl.plot_samples(problem=problem, filename='fig.png')
import os
os.remove('fig.png')
def test_plot_samples_3d():
problem = FooProblem3D()
problem.discretise_domain(n=10, mode='grid', variables = ['x', 'y', 'z'], locations=['D'])
pl = Plotter()
pl.plot_samples(problem=problem, filename='fig.png')
import os
os.remove('fig.png')
"""

View File

@@ -2,7 +2,7 @@ import torch
from pina.problem import AbstractProblem
from pina.condition import InputOutputPointsCondition
from pina.problem.zoo.supervised_problem import SupervisedProblem
from pina import RadiusGraph
from pina.graph import RadiusGraph
def test_constructor():
input_ = torch.rand((100,10))

View File

@@ -1,7 +1,7 @@
import torch
import pytest
from pina import TorchOptimizer, TorchScheduler
from pina.optim import TorchOptimizer, TorchScheduler
opt_list = [
torch.optim.Adam,