Using matplotlib2tikz to generate tikz figures, and using them in latex

main.tex

\documentclass{article}

\usepackage[utf8]{inputenc}
\usepackage{tikz}
\usepackage{pgfplots}

\begin{document}

\begin{figure}[ht]
  \centering
  \resizebox{6cm}{!}{\input{test.tex}}
  \caption{Testing}
  \label{fig:1}
\end{figure}

\end{document}

test.py

import numpy as np
import matplotlib.pyplot as plt

N = 1000
x = np.random.normal(scale=2, size=N)
y = 0.6*x + np.random.normal(scale=0.5, size=N)
plt.scatter(x, y, alpha=0.2, label='One')

x = np.random.normal(scale=2, size=N)
y = -0.2*x + np.random.normal(scale=0.5, size=N)
plt.scatter(x, y, alpha=0.2, label='Two')

plt.xlabel('x')
plt.ylabel('y')
plt.legend()

from matplotlib2tikz import save as tikz_save
tikz_save('test.tex')