switching to pytest

import unittest

import pytest

from common.mixins.multiple import MultiDatabase

class TestMultiDatabase(unittest.TestCase):

    def setUp(self):

        self.infl_id = "influx_id"

        self.influx = {"type": "influx", "id": self.infl_id, "host": "influx", "user": 'influx',

@pytest.fixture

def multi_database_fixture():

    infl_id = "influx_id"

    influx = {"type": "influx", "id": infl_id, "host": "influx", "user": 'influx',

              "password": 'secret', "database": "something", "port": '8086'}

        self.pg_id = "pg_db"

        self.postgres = {"type": "postgres", "id": self.pg_id, "host": "localhost", "user": 'me',

    pg_id = "pg_db"

    postgres = {"type": "postgres", "id": pg_id, "host": "localhost", "user": 'me',

                "password": 'nevergonnaguess', "database": "db", "port": '5435'}

        self.configs = [self.influx, self.postgres]

        self.multi = MultiDatabase(self.configs)

    configs = [influx, postgres]

    multi = MultiDatabase(configs)

    return multi, pg_id, infl_id

    def test_postgres_uri(self):

def test_postgres_uri(multi_database_fixture):

    multi, pg_id, infl_id = multi_database_fixture

    pg_expected_uri = 'postgresql://me:nevergonnaguess@localhost:5435/db'

        actual_pg_uri = self.multi.databases[self.pg_id].connection_info["uri"]

        self.assertEqual(actual_pg_uri, pg_expected_uri)

    actual_pg_uri = multi.databases[pg_id].connection_info["uri"]

    assert actual_pg_uri == pg_expected_uri

    def test_influx_uri(self):

def test_influx_uri(multi_database_fixture):

    multi, pg_id, infl_id = multi_database_fixture

    influx_expected_uri = 'influxdb://influx:secret@influx:8086/something'

        actual_influx_uri = self.multi.databases[self.infl_id].connection_info["uri"]

        self.assertEqual(actual_influx_uri, influx_expected_uri)

    actual_influx_uri = multi.databases[infl_id].connection_info["uri"]

    assert actual_influx_uri == influx_expected_uri

if __name__ == "__main__":

    unittest.main()

    pytest.main()