"""Tests for sme_data logic"""

from charts.logic import sme_data as sme_data


def test_get_sme_data_nfd_filter_false():
    response = sme_data.get_sme_data_nfd_filter(show_sme_data=False)

    assert response == [
        'N',
        'EditoriallySuspectContent',
    ]


def test_get_sme_data_nfd_filter_true():
    response = sme_data.get_sme_data_nfd_filter(show_sme_data=True)

    assert response == [
        'N',
        'EditoriallySuspectContent',
        'SMEAnalyticsDummy',
        'SwitchboardDummy',
    ]