# (kill-requests)$ python -m timeit -s 'import benchmark; benchmark.setup()' -n 2000 -r 10 -- "benchmark.test()"
2000 loops, best of 10: 1.63 msec per loop
# (master)$ python -m timeit -s 'import benchmark; benchmark.setup()' -n 2000 -r 10 -- "benchmark.test()"
2000 loops, best of 10: 1.82 msec per loop

from pynamodb.models import Model
from pynamodb.attributes import NumberAttribute, UnicodeAttribute


class UserModel(Model):
    class Meta:
        table_name = 'UserModel'
        host = 'http://localhost:8000'

    user_name = UnicodeAttribute(hash_key=True, attr_name='user_name')
    user_id = UnicodeAttribute(range_key=True)
    zip_code = NumberAttribute(null=True)
    email = UnicodeAttribute()


def setup():
    UserModel.create_table(read_capacity_units=1, write_capacity_units=1, wait=True)
    UserModel("test", user_id="1234", email="[email protected]", zip_code=1234).save()


def test():
    UserModel.get("test", "1234")