<reponame>terezaif/adventofcode
from aocd import get_data


def part1(data):
    depths = [s for s in data.split("\n")]

    return 150


def part2(data):
    depths = [s for s in data.split("\n")]

    return 900


def test_part1():
    assert part1(test_data) == 150


def test_part2():
    assert part2(test_data) == 900


data = get_data(day=2, year=2021)

print(part1(data))
print(part2(data))

test_data = """forward 5
down 5
forward 8
up 3
down 8
forward 2"""

from aocd import get_data


def part1(data):
    depths = [s for s in data.split("\n")]

    return 150


def part2(data):
    depths = [s for s in data.split("\n")]

    return 900


def test_part1():
    assert part1(test_data) == 150


def test_part2():
    assert part2(test_data) == 900


data = get_data(day=2, year=2021)

print(part1(data))
print(part2(data))

test_data = """forward 5
down 5
forward 8
up 3
down 8
forward 2"""
forward 5 down 5 forward 8 up 3 down 8 forward 2