Remove old test

Author: ndem0

tests/test_problem.py

@@ -48,25 +48,6 @@ def test_variables_correct_order_sampling():
         poisson_problem.input_variables
     )
 
-
-def test_input_pts():
-    n = 10
-    poisson_problem = Poisson()
-    poisson_problem.discretise_domain(n, "grid")
-    assert sorted(list(poisson_problem.input_pts.keys())) == sorted(
-        list(poisson_problem.conditions.keys())
-    )
-
-
-def test_collected_data():
-    n = 10
-    poisson_problem = Poisson()
-    poisson_problem.discretise_domain(n, "grid")
-    assert sorted(list(poisson_problem.collected_data.keys())) == sorted(
-        list(poisson_problem.conditions.keys())
-    )
-
-
 def test_add_points():
     poisson_problem = Poisson()
     poisson_problem.discretise_domain(1, "random", domains=["D"])
@@ -109,24 +90,4 @@ def test_wrong_custom_sampling_logic(mode):
 
     # Necessary cleanup
     if "new" in poisson_problem.domains:
-        del poisson_problem.domains["new"]
-
-
-def test_aggregate_data():
-    poisson_problem = Poisson()
-    poisson_problem.conditions["data"] = Condition(
-        input=LabelTensor(torch.tensor([[0.0, 1.0]]), labels=["x", "y"]),
-        target=LabelTensor(torch.tensor([[0.0]]), labels=["u"]),
-    )
-    poisson_problem.discretise_domain(1, "random", domains="all")
-    poisson_problem.collect_data()
-    assert isinstance(poisson_problem.collected_data, dict)
-    for name, conditions in poisson_problem.conditions.items():
-        assert name in poisson_problem.collected_data.keys()
-        if isinstance(conditions, InputTargetCondition):
-            assert "input" in poisson_problem.collected_data[name].keys()
-            assert "target" in poisson_problem.collected_data[name].keys()
-        elif isinstance(conditions, DomainEquationCondition):
-            assert "input" in poisson_problem.collected_data[name].keys()
-            assert "target" not in poisson_problem.collected_data[name].keys()
-            assert "equation" in poisson_problem.collected_data[name].keys()
+        del poisson_problem.domains["new"]