<filename>test/test_day_7.py
from test.helper import use_example


class part_1:
  @use_example(1)
  def test_example_1(self, problem):
    assert problem.solve_1() == 4

  def test_solution(self, problem):
    assert problem.solve_1() == 169


class part_2:
  @use_example(2)
  def test_example_2(self, problem):
    assert problem.solve_2() == 126

  def test_solution(self, problem):
    assert problem.solve_2() == 82372

<filename>test/test_day_7.py
from test.helper import use_example


class part_1:
  @use_example(1)
  def test_example_1(self, problem):
    assert problem.solve_1() == 4

  def test_solution(self, problem):
    assert problem.solve_1() == 169


class part_2:
  @use_example(2)
  def test_example_2(self, problem):
    assert problem.solve_2() == 126

  def test_solution(self, problem):
    assert problem.solve_2() == 82372
