from winton_kafka_streams.state.in_memory_key_value_store import InMemoryKeyValueStore


def test_inMemoryKeyValueStore():
    store = InMemoryKeyValueStore('teststore')

    store['a'] = 1
    assert store['a'] == 1

    store['a'] = 2
    assert store['a'] == 2

import pytest
from winton_kafka_streams.state.in_memory_key_value_store import InMemoryKeyValueStore


def test_inMemoryKeyValueStore():
    store = InMemoryKeyValueStore('teststore')

    store['a'] = 1
    assert store['a'] == 1

    store['a'] = 2
    assert store['a'] == 2

    del store['a']
    assert store.get('a') is None
    with pytest.raises(KeyError):
        store['a']

Test behaviour of key deletion
Test behaviour of key deletion

+ import pytest
  from winton_kafka_streams.state.in_memory_key_value_store import InMemoryKeyValueStore
  
  
  def test_inMemoryKeyValueStore():
      store = InMemoryKeyValueStore('teststore')
  
      store['a'] = 1
      assert store['a'] == 1
  
      store['a'] = 2
      assert store['a'] == 2
  
+     del store['a']
+     assert store.get('a') is None
+     with pytest.raises(KeyError):
+         store['a']
+ 
Test behaviour of key deletion
## Code Before:
from winton_kafka_streams.state.in_memory_key_value_store import InMemoryKeyValueStore


def test_inMemoryKeyValueStore():
    store = InMemoryKeyValueStore('teststore')

    store['a'] = 1
    assert store['a'] == 1

    store['a'] = 2
    assert store['a'] == 2

## Instruction:
Test behaviour of key deletion
## Code After:
import pytest
from winton_kafka_streams.state.in_memory_key_value_store import InMemoryKeyValueStore


def test_inMemoryKeyValueStore():
    store = InMemoryKeyValueStore('teststore')

    store['a'] = 1
    assert store['a'] == 1

    store['a'] = 2
    assert store['a'] == 2

    del store['a']
    assert store.get('a') is None
    with pytest.raises(KeyError):
        store['a']
