Exemplo n.º 1
0
def test_recording():
    stats = RecordingStatistics()
    with pytest.raises(TypeError):
        stats.record_entering()
    with pytest.raises(TypeError):
        stats.record_leaving()
    stats.wall = lambda: 10
    stats.record_starting(0)
    code = mock_code('foo')
    stat = RecordingStat(code)
    assert stat.name == 'foo'
    assert stat.calls == 0
    assert stat.total_time == 0
    stat.record_entering(100)
    stat.record_leaving(200)
    assert stat.calls == 1
    assert stat.total_time == 100
    stat.record_entering(200)
    stat.record_leaving(400)
    assert stat.calls == 2
    assert stat.total_time == 300
    code2 = mock_code('bar')
    stat2 = RecordingStat(code2)
    assert code2 not in stat
    stat.add_child(code2, stat2)
    assert code2 in stat
    assert stat.get_child(code2) is stat2
    assert len(stat) == 1
    assert list(stat) == [stat2]
    assert stat.total_time == 300
    assert stat.own_time == 300
    stat2.record_entering(1000)
    stat2.record_leaving(1004)
    assert stat2.total_time == 4
    assert stat2.own_time == 4
    assert stat.total_time == 300
    assert stat.own_time == 296
    stat.clear()
    assert len(stat) == 0
    with pytest.raises(TypeError):
        pickle.dumps(stat)
    stat3 = stat.ensure_child(mock_code('baz'))
    assert isinstance(stat3, VoidRecordingStat)
    stats.wall = lambda: 2000
    stats.record_stopping(400)
    assert stats.cpu_time == 400
    assert stats.wall_time == 1990
    assert stats.cpu_usage == 400 / 1990.
Exemplo n.º 2
0
def test_sorting():
    stat = RecordingStat(mock_code('foo'))
    stat1 = RecordingStat(mock_code('bar'))
    stat2 = RecordingStat(mock_code('baz'))
    stat3 = RecordingStat(mock_code('qux'))
    stat.add_child(stat1.code, stat1)
    stat.add_child(stat2.code, stat2)
    stat.add_child(stat3.code, stat3)
    stat.total_time = 100
    stat1.total_time = 20
    stat1.calls = 3
    stat2.total_time = 30
    stat2.calls = 2
    stat3.total_time = 40
    stat3.calls = 4
    assert stat.sorted() == [stat3, stat2, stat1]
    assert stat.sorted(by_calls) == [stat3, stat1, stat2]
    assert stat.sorted(by_total_time_per_call) == [stat2, stat3, stat1]
    assert stat.sorted(by_name) == [stat1, stat2, stat3]
Exemplo n.º 3
0
def test_frozen():
    code = mock_code('foo')
    stat = RecordingStat(code)
    stat.record_entering(0)
    stat.record_leaving(10)
    assert stat.name == 'foo'
    assert stat.total_time == 10
    frozen_stat = FrozenStat(stat)
    with pytest.raises(AttributeError):
        frozen_stat.record_entering(0)
    assert frozen_stat.name == 'foo'
    assert frozen_stat.total_time == 10
    restored_frozen_stat = pickle.loads(pickle.dumps(frozen_stat))
    assert restored_frozen_stat.name == 'foo'
    assert restored_frozen_stat.total_time == 10