Remove concept of unknown object type

1 files changed, 3 insertions, 8 deletions

diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py
index 8c19954..899a776 100644
--- a/tests/test_memory_store.py
+++ b/tests/test_memory_store.py
@@ -15,18 +15,13 @@ class TestMemoryStore(unittest.TestCase):
         oid = 'https://test/object/12345'
         obj = objects.Object(id=oid)
         self.store.add(obj)
-        self.assertIn('object', self.store.objects)
-        self.assertIn(oid, self.store.objects['object'])
+        self.assertIn(oid, self.store.objects)
 
     def test_get_object(self):
-        res = self.store.get('object', self.oid)
+        res = self.store.get(self.oid)
         self.assertEqual(res, self.obj)
 
     def test_get_object_not_existing(self):
         oid = 'https://test/object/does_not_exist'
         with self.assertRaises(exceptions.DoesNotExist):
-            self.store.get('object', oid)
-
-    def test_get_object_wrong_type(self):
-        with self.assertRaises(exceptions.UnknownObjectType):
-            self.store.get('no_such_type', self.oid)
+            self.store.get(oid)