import unittest

from ..inmemory import (
    DataValueInMemory, DataStreamInMemory, DataStoreNestableInMemory)
from .mixintestcase import (
    MixInValueTestCase, MixInStreamTestCase, MixInNestableAutoValueTestCase)


class TestDataValueInMemory(MixInValueTestCase, unittest.TestCase):

    dstype = DataValueInMemory

    def test_set_get_singleton(self):
        obj = object()
        self.ds.set(obj)
        self.assertTrue(self.ds.get() is obj)


class TestDataStreamInMemory(MixInStreamTestCase, unittest.TestCase):
    dstype = DataStreamInMemory


class TestDataStoreNestableInMemory(MixInNestableAutoValueTestCase,
                                    unittest.TestCase):
    dstype = DataStoreNestableInMemory

import unittest

from ..inmemory import (
    DataValueInMemory, DataStreamInMemory,
    DataStoreNestableInMemory, DataStoreNestableInMemoryAutoValue)
from .mixintestcase import (
    MixInValueTestCase, MixInStreamTestCase,
    MixInNestableTestCase, MixInNestableAutoValueTestCase)


class TestDataValueInMemory(MixInValueTestCase, unittest.TestCase):

    dstype = DataValueInMemory

    def test_set_get_singleton(self):
        obj = object()
        self.ds.set(obj)
        self.assertTrue(self.ds.get() is obj)


class TestDataStreamInMemory(MixInStreamTestCase, unittest.TestCase):
    dstype = DataStreamInMemory


class TestDataStoreNestableInMemory(MixInNestableTestCase,
                                    unittest.TestCase):
    dstype = DataStoreNestableInMemory


class TestDataStoreNestableInMemoryAutoValue(MixInNestableAutoValueTestCase,
                                             unittest.TestCase):
    dstype = DataStoreNestableInMemoryAutoValue

Fix and add tests for datastore.inmemory
Fix and add tests for datastore.inmemory

  import unittest
  
  from ..inmemory import (
-     DataValueInMemory, DataStreamInMemory, DataStoreNestableInMemory)
+     DataValueInMemory, DataStreamInMemory,
+     DataStoreNestableInMemory, DataStoreNestableInMemoryAutoValue)
  from .mixintestcase import (
+     MixInValueTestCase, MixInStreamTestCase,
-     MixInValueTestCase, MixInStreamTestCase, MixInNestableAutoValueTestCase)
+     MixInNestableTestCase, MixInNestableAutoValueTestCase)
  
  
  class TestDataValueInMemory(MixInValueTestCase, unittest.TestCase):
  
      dstype = DataValueInMemory
  
      def test_set_get_singleton(self):
          obj = object()
          self.ds.set(obj)
          self.assertTrue(self.ds.get() is obj)
  
  
  class TestDataStreamInMemory(MixInStreamTestCase, unittest.TestCase):
      dstype = DataStreamInMemory
  
  
- class TestDataStoreNestableInMemory(MixInNestableAutoValueTestCase,
+ class TestDataStoreNestableInMemory(MixInNestableTestCase,
                                      unittest.TestCase):
      dstype = DataStoreNestableInMemory
  
+ 
+ class TestDataStoreNestableInMemoryAutoValue(MixInNestableAutoValueTestCase,
+                                              unittest.TestCase):
+     dstype = DataStoreNestableInMemoryAutoValue
+ 
Fix and add tests for datastore.inmemory
## Code Before:
import unittest

from ..inmemory import (
    DataValueInMemory, DataStreamInMemory, DataStoreNestableInMemory)
from .mixintestcase import (
    MixInValueTestCase, MixInStreamTestCase, MixInNestableAutoValueTestCase)


class TestDataValueInMemory(MixInValueTestCase, unittest.TestCase):

    dstype = DataValueInMemory

    def test_set_get_singleton(self):
        obj = object()
        self.ds.set(obj)
        self.assertTrue(self.ds.get() is obj)


class TestDataStreamInMemory(MixInStreamTestCase, unittest.TestCase):
    dstype = DataStreamInMemory


class TestDataStoreNestableInMemory(MixInNestableAutoValueTestCase,
                                    unittest.TestCase):
    dstype = DataStoreNestableInMemory

## Instruction:
Fix and add tests for datastore.inmemory
## Code After:
import unittest

from ..inmemory import (
    DataValueInMemory, DataStreamInMemory,
    DataStoreNestableInMemory, DataStoreNestableInMemoryAutoValue)
from .mixintestcase import (
    MixInValueTestCase, MixInStreamTestCase,
    MixInNestableTestCase, MixInNestableAutoValueTestCase)


class TestDataValueInMemory(MixInValueTestCase, unittest.TestCase):

    dstype = DataValueInMemory

    def test_set_get_singleton(self):
        obj = object()
        self.ds.set(obj)
        self.assertTrue(self.ds.get() is obj)


class TestDataStreamInMemory(MixInStreamTestCase, unittest.TestCase):
    dstype = DataStreamInMemory


class TestDataStoreNestableInMemory(MixInNestableTestCase,
                                    unittest.TestCase):
    dstype = DataStoreNestableInMemory


class TestDataStoreNestableInMemoryAutoValue(MixInNestableAutoValueTestCase,
                                             unittest.TestCase):
    dstype = DataStoreNestableInMemoryAutoValue
