personal-data/test/test_csv_import.py

import datetime

import frozendict

from personal_data.csv_import import determine_possible_keys


def test_determine_possible_keys():
    data = frozendict.frozendict(
        {
            'game.name': 'Halo',
            'me.last_played_time': datetime.datetime(
                2021, 6, 13, 19, 12, 21, tzinfo=datetime.timezone.utc,
            ),
            'trophy.name': 'Test',
            'trophy.desc': 'Description',
        },
    )
    keys = determine_possible_keys(data)

    assert keys.time_start == []
    assert keys.time_end == ['me.last_played_time']
    assert keys.duration == []
    assert len(keys.name) == 3