def test_acquire(self): nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(CONFIG) with mock.patch.object(newt.Newt, "get_usage") as f: f.return_value = utils.input_from_file(ALLOCATIONS_FIXTURE_FILE) res = nersc_allocations.acquire() assert PRODUCES == list(res.keys()) assert EXPECTED_PANDAS_DFRAME.equals(res[PRODUCES[0]])
def test_acquire(self): nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(config) with mock.patch.object(newt.Newt, 'get_usage') as f: f.return_value = utils.input_from_file( 'newt_allocations.cs.fixture') res = nersc_allocations.acquire() assert produces == res.keys() assert expected_pandas_dframe.equals(res[produces[0]])
def test_acquire(): def side_effect_get_usage(username): if username == FAKE_USER: return {"items": []} return utils.input_from_file(ALLOCATIONS_FIXTURE_FILE) nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(CONFIG) with mock.patch.object(newt.Newt, "get_usage") as f: f.side_effect = side_effect_get_usage res = nersc_allocations.acquire() verify_products(nersc_allocations, res) assert EXPECTED_PANDAS_DFRAME.equals(res["Nersc_Allocation_Info"])
def test_acquire(self): def side_effect_get_usage(username): if username == FAKE_USER: return {'items': []} return utils.input_from_file(ALLOCATIONS_FIXTURE_FILE) nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(CONFIG) with mock.patch.object(newt.Newt, "get_usage") as f: f.side_effect = side_effect_get_usage res = nersc_allocations.acquire() assert PRODUCES == list(res.keys()) assert EXPECTED_PANDAS_DFRAME.equals(res[PRODUCES[0]])
def test_produces(): nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(CONFIG) assert nersc_allocations._produces == _PRODUCES
def test_produces(self): nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(CONFIG) assert nersc_allocations.produces() == PRODUCES
def test_produces(self): nersc_allocations = NerscAllocationInfo.NerscAllocationInfo(config) assert nersc_allocations.produces() == produces