# -*- coding: utf-8 -*-
from transformers.opendata.operations_stats import OperationsStatsTransformer
from opendata.base import BaseTest


class TestOperationsStatsTransformer(BaseTest):
    def test_columns(self):
        self.assertEquals(
            OperationsStatsTransformer("/tmp/fake").columns_to_drop(),
            [
                "nombre_navires_mas_omi",
                "concerne_snosan",
                "nombre_moyens_nautiques_engages",
                "nombre_moyens_terrestres_engages",
                "nombre_moyens_aeriens_engages",
                "duree_engagement_moyens_nautiques_minutes",
                "duree_engagement_moyens_terrestres_minutes",
                "duree_engagement_moyens_aeriens_minutes",
                "duree_engagement_moyens_nautiques_heures",
                "duree_engagement_moyens_terrestres_heures",
                "duree_engagement_moyens_aeriens_heures",
            ],
        )

    def transformers(self):
        return [None, OperationsStatsTransformer]

    def test_model(self):
        self.model_is("OperationStats")
