Exemple #1
0
def test_production_volume_keyerror():
    dataset = {
        'exchanges': [{
            'type': 'reference product',
        }]
    }
    assert production_volume(dataset) is None
Exemple #2
0
def test_production_volume_no_amount():
    dataset = {
        'exchanges': [{
            'type': 'reference product',
            'production volume': {}
        }]
    }
    assert production_volume(dataset) is None
Exemple #3
0
def test_production_volume_multiple():
    dataset = {
        'exchanges': [{
            'type': 'reference product',
        }, {
            'type': 'reference product',
        }]
    }
    assert production_volume(dataset) is None
Exemple #4
0
def test_production_volume():
    dataset = {
        'exchanges': [{
            'type': 'something else',
            'production volume': {
                'amount': 2
            },
        }, {
            'type': 'reference product',
            'production volume': {
                'amount': 42
            },
        }]
    }
    assert production_volume(dataset) == 42
Exemple #5
0
def test_production_volume_none():
    dataset = {'exchanges': []}
    assert production_volume(dataset) is None
Exemple #6
0
def test_update_market_production_volumes_recursive():
    given = [{
        'name':
        '',
        'code':
        'a',
        'type':
        'market group',
        'location':
        'GLO',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [
            {
                'code': 'b'
            },
            {
                'code': 'd'
            },
        ]
    }, {
        'name':
        '',
        'code':
        'b',
        'type':
        'market group',
        'location':
        'NAFTA',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [{
            'code': 'c'
        }, {
            'production volume': {
                'amount': 10
            },
            'code': 'e'
        }]
    }, {
        'name':
        '',
        'code':
        'c',
        'type':
        'market group',
        'location':
        'US',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [{
            'production volume': {
                'amount': 20
            },
            'code': 'f'
        }]
    }, {
        'name':
        '',
        'code':
        'd',
        'type':
        'market group',
        'location':
        'CN',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [{
            'production volume': {
                'amount': 30
            },
            'code': 'g'
        }]
    }]
    update_market_production_volumes(given, 'market group')
    assert production_volume(given[0]) == 60
    assert production_volume(given[1]) == 30
    assert production_volume(given[2]) == 20
    assert production_volume(given[3]) == 30
Exemple #7
0
def test_update_market_production_volumes_recursive():
    given = [{
        'name': '',
        'code': 'a',
        'type': 'market group',
        'location': 'GLO',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [
            {'code': 'b'},
            {'code': 'd'},
        ]
    }, {
        'name': '',
        'code': 'b',
        'type': 'market group',
        'location': 'NAFTA',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [
            {'code': 'c'},
            {'production volume': {'amount': 10}, 'code': 'e'}
        ]
    }, {
        'name': '',
        'code': 'c',
        'type': 'market group',
        'location': 'US',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [
            {'production volume': {'amount': 20}, 'code': 'f'}
        ]
    }, {
        'name': '',
        'code': 'd',
        'type': 'market group',
        'location': 'CN',
        'exchanges': [{
            'name': '',
            'type': 'reference product',
            'production volume': {},
        }],
        'suppliers': [
            {'production volume': {'amount': 30}, 'code': 'g'}
        ]
    }]
    update_market_production_volumes(given, 'market group')
    assert production_volume(given[0]) == 60
    assert production_volume(given[1]) == 30
    assert production_volume(given[2]) == 20
    assert production_volume(given[3]) == 30