Python Util.get_target_partitions_stringの例

コード例 #1

ファイル: test_util.py プロジェクト: adidas/m3d-api

    def test_get_target_partitions_string(self):
        """
        This method tests the correct functionality of get_partition_column_string of Util class
        :return:
        """
        assert Util.get_target_partitions_string("year") == "year"
        assert Util.get_target_partitions_string("month") == "year,month"
        assert Util.get_target_partitions_string("day") == "year,month,day"
        assert Util.get_target_partitions_string("") == ""

        assert Util.get_target_partitions_string("country") == "country"

コード例 #2

    def test_get_target_partitions_string(self):
        """
        This method tests the correct functionality of get_partition_column_string of Util class
        :return:
        """
        assert Util.get_target_partitions_string("year") == "year"
        assert Util.get_target_partitions_string("month") == "year,month"
        assert Util.get_target_partitions_string("day") == "year,month,day"
        assert Util.get_target_partitions_string("") == ""

        with pytest.raises(Exception) as exc_info:
            Util.get_target_partitions_list("country")
        assert "Partition type country not supported" in str(exc_info.value)