def test_set_consecutive_before_eq(): """Test setting consecutive ranges to the same value.""" rm = RangeMap({1: 'a', 2: 'b', 3: 'c'}) print_underlying(rm) rm.set('b', 1, 2) print_underlying(rm) assert rm == RangeMap({1: 'b', 3: 'c'})
def test_set_beg(): """Test setting the beginning.""" rm = RangeMap() rm.set('a', stop=4) with pytest.raises(KeyError): rm[4] assert rm[3] == 'a'
def test_break_up_existing_internal_interval(): """Test breaking up an existing interval.""" rm = RangeMap({1: 'a', 2: 'b'}) rm.set('d', start=1, stop=1.5) assert rm[1] == 'd' assert rm[1.5] == 'a' assert rm[2] == 'b' assert rm[3] == 'b'
def test_break_up_existing_open_end_interval(): """Test breaking up an existing open interval at the end.""" rm = RangeMap({1: 'a', 2: 'b'}) rm.set('d', start=2, stop=2.5) assert rm[1] == 'a' assert rm[2] == 'd' assert rm[2.5] == 'b' assert rm[3] == 'b'
def test_set_closed_interval_end(): """Test setting a closed range on the end.""" rm = RangeMap({1: 'a', 2: 'b'}) rm.set('c', start=3, stop=4) assert rm[1] == 'a' assert rm[2] == 'b' assert rm[3] == 'c' assert rm[4] == 'b'
def test_overwrite_multiple_internal(): """Test overwriting multiple adjoining intervals.""" rm = RangeMap({1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'}) rm.set('z', start=2, stop=5) assert rm[1] == 'a' assert rm[2] == 'z' assert rm[3] == 'z' assert rm[4] == 'z' assert rm[5] == 'e'
def test_dates(): """Test using dates.""" rm = RangeMap() rm.set('b', datetime.date(1936, 12, 11)) rm.set('a', datetime.date(1952, 2, 6)) assert rm[datetime.date(1945, 1, 1)] == 'b' assert rm[datetime.date(1965, 4, 6)] == 'a' with pytest.raises(KeyError): rm[datetime.date(1900, 1, 1)]
def test_overwrite_all(): """Test overwriting the entire mapping.""" rm = RangeMap({1: 'a', 2: 'b'}) rm.set('z', start=0) with pytest.raises(KeyError): rm[-1] assert rm[0] == 'z' assert rm[1] == 'z' assert rm[2] == 'z' assert rm[3] == 'z'
def test_set_existing_interval(): """Test setting an exact existing range.""" rm = RangeMap({1: 'a', 2: 'b'}) rm.set('c', start=1, stop=2) print_underlying(rm) assert rm[1] == 'c' assert rm[2] == 'b' assert rm[3] == 'b' assert rm == RangeMap({1: 'c', 2: 'b'}) with pytest.raises(KeyError): rm[0]
def test_closed(): """Test a closed RangeMap.""" rm = RangeMap() rm.set('a', start=1, stop=2) print_underlying(rm) assert rm[1] == 'a' assert rm[1.9] == 'a' with pytest.raises(KeyError): rm[2] with pytest.raises(KeyError): rm[0]
def test_default_value(): """Test setting just a default value.""" rm = RangeMap(default_value=None) print(rm) assert rm[1] is None assert rm[-2] is None rm.set('a', start=1) print(rm) assert rm[0] is None assert rm[1] == 'a' assert rm[2] == 'a'
def test_alter_beg(): """Test altering the beginning.""" rm = RangeMap({1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'}) rm.set('z', stop=3) assert rm[0] == 'z' assert rm[1] == 'z' assert rm[2] == 'z' assert rm[3] == 'c' assert rm[4] == 'd' assert rm[5] == 'e' rm.set('y', stop=3) assert rm == RangeMap({3: 'c', 4: 'd', 5: 'e'}, default_value='y')
def test_simple_set(): """Test set.""" rm = RangeMap() rm.set('a', start=1) print_underlying(rm) assert rm[1] == 'a' assert rm[2] == 'a' with pytest.raises(KeyError): rm[0] rm.set('b', start=2) assert rm[1] == 'a' assert rm[2] == 'b' assert rm[3] == 'b'
def test_set_consecutive_after_eq(): """Test setting consecutive ranges to the same value.""" rm = RangeMap({1: 'a', 2: 'b', 3: 'c'}) rm.set('a', 2, 3) assert rm == RangeMap({1: 'a', 3: 'c'})
def test_start_gt_stop(): rm = RangeMap({1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'}, default_value='z') with pytest.raises(ValueError): rm.set('a', start=3, stop=2) with pytest.raises(ValueError): rm.get_range(start=3, stop=2)
def test_from_mapping(): """Test creating a RangeMap from a mapping.""" rm = RangeMap() rm.set('a', start=1) rm.set('b', start=2) assert rm == RangeMap({1: 'a', 2: 'b'})
def test_whole_range(): """Test setting the whole range.""" rm = RangeMap() rm.set('a') assert rm[1] == 'a' assert rm[-1] == 'a'
def test_set_consecutive_between_eq(): """Test setting consecutive ranges to the same value.""" rm = RangeMap({1: 'a', 2: 'b', 3: 'c', 4: 'b'}) rm.set('b', 3, 4) assert rm == RangeMap({1: 'a', 2: 'b'})