from wallace import models, memes, db


class TestMemes(object):

    def setup(self):
        self.db = db.init_db(drop_all=True)

    def teardown(self):
        self.db.rollback()
        self.db.close()

    def add(self, *args):
        self.db.add_all(args)
        self.db.commit()

    def test_create_genome(self):
        node = models.Node()
        meme = memes.Genome(origin=node)
        self.add(node, meme)

        assert meme.origin_id == node.id
        assert meme.type == "genome"
        assert meme.contents is None

    def test_create_mimeme(self):
        node = models.Node()
        meme = memes.Mimeme(origin=node)
        self.add(node, meme)

        assert meme.origin_id == node.id
        assert meme.type == "mimeme"
        assert meme.contents is None

from wallace import models, memes, db


class TestMemes(object):

    def setup(self):
        self.db = db.init_db(drop_all=True)

    def teardown(self):
        self.db.rollback()
        self.db.close()

    def add(self, *args):
        self.db.add_all(args)
        self.db.commit()

    def test_create_genome(self):
        meme = memes.Genome()
        self.add(meme)

        assert meme.type == "genome"
        assert meme.contents is None

    def test_create_mimeme(self):
        meme = memes.Mimeme()
        self.add(meme)

        assert meme.type == "mimeme"
        assert meme.contents is None

Remove origin from meme model
Remove origin from meme model

  from wallace import models, memes, db
  
  
  class TestMemes(object):
  
      def setup(self):
          self.db = db.init_db(drop_all=True)
  
      def teardown(self):
          self.db.rollback()
          self.db.close()
  
      def add(self, *args):
          self.db.add_all(args)
          self.db.commit()
  
      def test_create_genome(self):
-         node = models.Node()
-         meme = memes.Genome(origin=node)
+         meme = memes.Genome()
-         self.add(node, meme)
+         self.add(meme)
  
-         assert meme.origin_id == node.id
          assert meme.type == "genome"
          assert meme.contents is None
  
      def test_create_mimeme(self):
-         node = models.Node()
-         meme = memes.Mimeme(origin=node)
+         meme = memes.Mimeme()
-         self.add(node, meme)
+         self.add(meme)
  
-         assert meme.origin_id == node.id
          assert meme.type == "mimeme"
          assert meme.contents is None
  
Remove origin from meme model
## Code Before:
from wallace import models, memes, db


class TestMemes(object):

    def setup(self):
        self.db = db.init_db(drop_all=True)

    def teardown(self):
        self.db.rollback()
        self.db.close()

    def add(self, *args):
        self.db.add_all(args)
        self.db.commit()

    def test_create_genome(self):
        node = models.Node()
        meme = memes.Genome(origin=node)
        self.add(node, meme)

        assert meme.origin_id == node.id
        assert meme.type == "genome"
        assert meme.contents is None

    def test_create_mimeme(self):
        node = models.Node()
        meme = memes.Mimeme(origin=node)
        self.add(node, meme)

        assert meme.origin_id == node.id
        assert meme.type == "mimeme"
        assert meme.contents is None

## Instruction:
Remove origin from meme model
## Code After:
from wallace import models, memes, db


class TestMemes(object):

    def setup(self):
        self.db = db.init_db(drop_all=True)

    def teardown(self):
        self.db.rollback()
        self.db.close()

    def add(self, *args):
        self.db.add_all(args)
        self.db.commit()

    def test_create_genome(self):
        meme = memes.Genome()
        self.add(meme)

        assert meme.type == "genome"
        assert meme.contents is None

    def test_create_mimeme(self):
        meme = memes.Mimeme()
        self.add(meme)

        assert meme.type == "mimeme"
        assert meme.contents is None
